在使用 pytorch 保存检查点的同时训练失速

如何解决在使用 pytorch 保存检查点的同时训练失速

我正在尝试在 google colab 中使用 pytorch 时保存检查点。 在所有时代完成后保存它时 - 它会成功保存。但是，当我使用下面的代码将它保存在 for 循环中时，训练将永远停止并且指针卡在 torch.save(checkpoint,....) 行上。一点帮助将非常有用。 如果 np.mean(tmp_eval_rmse)

问候， Mithun Thakkar。