torch>=1.3.0
tqdm
pandas
numpy
torchsummaryX>=2.0
pytorch_lightning==0.6.0
tensorboardX
optuna
