記錄,pytorch v1.4 似乎有嚴重bug
pytorch v1.4 cuda101 windows
訓練複雜模型時,例如DeepLabV3,收斂會非常困難,猜測是內部梯度計算有錯誤。
降級到 v1.3.1 cuda101 windows,收斂變的正常
不知道其他版本有沒有這樣的問題,但如果發現收斂困難,請降級到v1.3.1。
pytorch v1.4 嚴重收斂bug
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
記錄,pytorch v1.4 似乎有嚴重bug
pytorch v1.4 cuda101 windows
訓練複雜模型時,例如DeepLabV3,收斂會非常困難,猜測是內部梯度計算有錯誤。
降級到 v1.3.1 cuda101 windows,收斂變的正常
不知道其他版本有沒有這樣的問題,但如果發現收斂困難,請降級到v1.3.1。