geomloss
torch
scikit-learn
numpy
pandas
tqdm
pyyaml
torchvision
matplotlib
seaborn