imageio==2.20.0
matplotlib==3.1.2
numpy==1.17.4
scikit_learn==1.2.2
torch==1.10.0+cu111
torchinfo==1.6.1
torchvision==0.11.0+cu111
tqdm==4.62.3
