torch
torchvision
pytorch-lightning
simple-parsing
albumentations
matplotlib