我正在使用 Pytorch，我可以在 Colab GPU 上训练我的模型。我的模型不是很大。大概是10M的参数。当我尝试在 Colab TPU 上训练时，会话崩溃并突然重新启动。我尝试了每晚和默认版本的 XLA 。我检查了内存没有增加。
感谢您的帮助
暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！
如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com (将#修改为@）
