Blame · examples/pytorch/image-pretraining/requirements.txt · 6c7b68d414cd170c0efd3b745a924cbef6e2a5d8 · chenpangpang / transformers · GitLab

Find file Normal view History Permalink

requirements.txt 47 Bytes

Newer Older

[ViTMAE] Add image pretraining script (#15242)

NielsRogge committed Jan 21, 2022

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