torch>=1.2.0
torchvision
wandb
argparse
sklearn
numpy
python-utils
keras
pytorch-lightning