簡體   English   中英

為什么在 google colab 上訓練的 model 權重不能像在 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.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM