Out of space

Hi

i tried to run the below code(pre trained resnet34) and got out of memory error as shown in image. Can I download the required pre-trained models to a specific folder instead of home folder(as my home folder is just 1GB).

Hi Dinesh,
first, as your picture showed, you downloaded successfully the pretrained model because it printed “100%”.
So your problem must be somewhere else. Your screen shot is complete, please show us the last lines of the error stack message to know further about the problem.

Hi Iskode,
here are the screen shots,

@timiit, what’s your batch size? How much memory does your GPU have?