bump pyTorch to 2.1+
Borda opened this issue · 0 comments
Borda commented
🚀 Feature
atm Kaggle runs toch 2.0, but as 2.1 was released some time ago, it would be great to have an upgrade
Motivation
Use the latest torch features mainly for TorchVision augmentations
Additional context
I may have a look and contribute 🦩