[英]Importing Locally Trained TensorFlow Model to Google Colab
[英]Why model weights trained on google colab cannot perform locally as in colab
最近我在 google colab 上訓練了我的 model,並通過 ModelCheckPoint 自動將權重保存在 google drive 中。
當我在 google colab 上使用這些保存的權重時,預測效果很好,但是當我下載權重並在筆記本電腦上本地運行它們時,預測非常糟糕。
我的筆記本電腦的 GPU 是 GeForce RTX 1060,Max-Q 設計 12GB。 誰能告訴我是什么導致我的筆記本電腦和 google colab 之間的性能不同?
順便說一句,我嘗試保存整個 model 然后加載它,保存過程運行順利,但是當我加載 model 時,有一個ValueError說“函數之間存在循環依賴” ,仍然不知道如何解決它。
我通過將 tensorflow-gpu==2.2.0 更改為 tensorflow==2.3.0 解決了這個問題,這與 google colab 中的版本完全相同。
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.