swanlab
accelerate
torchvision
