transformers
6c7b68d4 - [ViTMAE] Add image pretraining script (#15242)

Commit
4 years ago
[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
Author
Parents
Loading