numpy
torch
torchmetrics

[colorama]
colorama>=0.4.3

[mlflow]
mlflow>=1.12.1

[pandas]
pandas>=2.0.0.0

[scikit-learn]
scikit-learn>=0.23.2

[tensorboard]
tensorboard>=2.4.0

[tensorboardX]
tensorboardX>=2.1

[torchvision]
torchvision>=0.8.1
