torch>=1.7
torchmetrics
pytorch-lightning>=1.3.1
PyYAML>=5.1
numpy
pandas
packaging
tqdm
