colab out of ram session crashed

[AK391]

after running a couple times on different images and changing

standard PyTorch mean-std input image normalization

transform = T.Compose([
T.Resize(800),
T.ToTensor(),
T.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
])

to

standard PyTorch mean-std input image normalization

transform = T.Compose([
T.Resize(500),
T.ToTensor(),
T.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
])

I get session crashed

[AK391]

@alcinos @ashkamath is there a place where it is hogging up memory?