Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "2fb8ddeefff6c9eaa8aed4345013d094cf413547"
Switch branch/tag
Find file BlameHistoryPermalink
  • NielsRogge's avatar
    [ViTMAE] Add image pretraining script (#15242) · 6c7b68d4
    NielsRogge authored 
    * 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
    6c7b68d4
test_examples.py 20.3 KB