torch==1.8.0
torchvision==0.9.0
pyyaml==5.4.1
