关于google colab 的一个问题

Question

问题描述：我在colab 上面跑了一个测试，本地Tensorflow与colab 上的版本保持一致，本地与colab上面均可以正常保存、导入 checkpoint 但是，我把colab上训练好的checkpoint 下载到本地再导入运行，结果跑飞了。
问题：怎样才能正确的保存模型，并在不同的平台之间迁移

Answer 1

正十七 2020-06-07 20:33:52

同学你好，看你的描述，感觉应该是正确的方式了。这里我觉得跑非的原因可能是learning rate的变化，比如原来训练checkpoint的时候learning rate调整成了1e-5, 然后重新载入以后可能变成了1e-3，所以才跑飞。

另外，感觉不一定是colab的问题，你逆向试一下，即在colab上训练，然后在本地载入，看会不会跑飞？如果也会，那么跟colab无关。

• 提问者 DATUONIAO #1

  非常感谢！