loguru
tqdm
pyyaml
wandb
torch
torchvision
