pytorch-lightning>=1.4.0
torch>=1.7.1
torch-model-archiver
