transformers/examples/pytorch/image-pretraining/requirements.txt
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

3 lines
47 B
Plaintext

torch>=1.5.0
torchvision>=0.6.0
datasets>=1.8.0