pytorch-image-models/requirements.txt

4 lines
39 B
Plaintext

torch>=1.4.0
torchvision>=0.5.0
pyyaml