torch
albumentations
torchvision