0

我正在使用 Pytorch，我可以在 Colab GPU 上训练我的模型。我的模型不是很大。大约是10M的参数。当我尝试在 Colab TPU 上进行训练时，会话突然崩溃并重新启动。我尝试了 XLA 的每晚版本和默认版本。我检查了它没有增加的RAM。

谢谢你的帮助

4

0 回答 0