NielsRogge
|
6c7b68d414
|
[ViTMAE] Add image pretraining script (#15242)
* Add script
* Improve script
* Fix data collator
* Update README
* Add label_names argument
* Apply suggestions from code review
* Add config parameters
* Update script
* Fix bug
* Improve README
* Improve README and add test
* Fix import
* Add image_column_name
|
2022-01-21 12:11:08 +01:00 |
|