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