torch>=1.12.*
torchmetrics==0.9.3
tensorboard
numpy
PyYAML
tqdm