nachiket273/pytorch_resnet_rs

Training resnet-rs

AleMaiaF opened this issue · 0 comments

Hello,

Thanl you for sharing your work. I am trying to train a resnet-rs152 model with fastAPI in kaggle but I keep receiving the "CUDA out of memory" warning when using the "learn.lr_find()" or training function (same with Colab). Is there a recomended method for training this model?