torch < 2
torchvision < 1
Pillow>=8.4.0, < 10
torchmetrics < 1
nltk
numpy
pyyaml
h5py