PyYAML>=5.1
numpy
pandas
torch
tensorboard
packaging
datasets
omegaconf