numpy==1.20.0
torch==1.9.0
torchaudio==0.9.0
torchvision==0.10.0
matplotlib==3.5.1
scikit-learn==1.0.2