关于google colab 的一个问题
来源:9-11 DCGAN实战引⼊

DATUONIAO
2020-06-05
问题描述:我在colab 上面跑了一个测试,本地Tensorflow与colab 上的版本保持一致,本地与colab上面均可以正常保存、导入 checkpoint 但是,我把colab上训练好的checkpoint 下载到本地再导入运行,结果跑飞了。
问题:怎样才能正确的保存模型,并在不同的平台之间迁移
写回答
1回答
-
同学你好,看你的描述,感觉应该是正确的方式了。这里我觉得跑非的原因可能是learning rate的变化,比如原来训练checkpoint的时候learning rate调整成了1e-5, 然后重新载入以后可能变成了1e-3,所以才跑飞。
另外,感觉不一定是colab的问题,你逆向试一下,即在colab上训练,然后在本地载入,看会不会跑飞?如果也会,那么跟colab无关。
012020-06-08
相似问题