Colab 在尝试创建混淆矩阵时崩溃

Question

我正在尝试为我的测试集创建一个混淆矩阵。我的测试集包含 3585 张图像。每当我尝试 运行 以下代码时：

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

Google colab 在使用所有可用 RAM 时崩溃。有人能解决这个问题吗？我应该分批进行吗？

Answer 1

Should I do this in batches?

是的！尝试减小批量大小。

dataloader = ...  # reduce batch size here on dataloader creation
...

y_pred = []
for batch in dataloader:
    batch_y_pred = resnet(batch)
    y_pred.append(batch_y_pred)

我用的是list with append，你可以换个方式试试