Also my images are 512x512 and i use a batch of 10 images(it uses the full colab gpu)and it takes like 7 minute per epoch.
So i reduce image to 256x256 or 128x128… but not the resize from pytorch which use interpolation, i take pixel from 4 to 4 on colums and rows (that for 128x128) or 2 by 2 (for 256x256) to train with a batch of 30. I dont think this is a big problem, this was recommended by a teacher.
And i used scheduler for adam lr, but if i use scheduler after every epcoch it doenst change the lr, but if i use after iterations its changes too much like lr = 0.0000005