torch==2.0.1
torchvision==0.15.2
matplotlib==3.7.2
pyyaml==6.0
numpy==1.24.1