jsonargparse
lightning[pytorch-extra]
matplotlib
mlflow
numpy
pandas
parse
pycocotools
pydot
pyyaml
scikit-learn
scipy
seaborn
tabulate
torch
torchmetrics
torchvision
tqdm
