torchvision<=0.4.0,>=0.2.2
pretrainedmodels==0.7.4

[test]
pytest
