PyYAML>=5.3.1
torchvision>=0.4.0
