pytorch-image-models/requirements.txt

4 lines
39 B
Plaintext

torch>=1.2.0
torchvision>=0.4.0
pyyaml