Colab crashes when trying to create confusion matrix

I am trying to create a confusion matrix for my test set. My test set consists of 3585 images. Whenever I try to run the following code:

x_test,y_test = next(iter(dataloader)))
y_pred = resnet(x_test)

Google colab crashes using all the available RAM. Does anyone have a work around for this? Should I do this in batches?



Comments

Popular posts from this blog

I get wrong characters when retreiving the message body of an email using TIdIMAP4.UIDRetrieveTextPeek2()

How to drop the all the 1's in a correlation matrix

Today Walkin 14th-Sept