Minor improvements

[leonardpasi]

• disable beta warning for torchvision transforms
• pre-commit hooks run successfully for all files
• localization models uploaded to wandb at the end of training run