clearml
pytorch_lightning ~= 1.1.2
torch
torchvision