Build SSD GPU 訓練問題
2020/06/05 下午 05:04
電腦視覺深度學習討論版
Vincent Tsou
觀看數:14
回答數:1
收藏數:0
cvdl
cvdl-d29
在執行 Day 29 Build SSD_VGG.ipynb 時遇到兩個問題,在基本設定的程式碼中原始程式碼如下
執行後會出現 args is not defined 的錯誤,改為下圖之程式碼後
這裡可以正常執行
但是確發生下列的錯誤
導致我無法使用 GPU 訓練模型,請問是那一個地方出了問題?可以如何修正 ?
回答列表
-
2020/06/06 上午 10:56楊哲寧贊同數:0不贊同數:0留言數:0
1.您好,感謝您自己修正第一個問題,這部分我們再檢查一下。
2.針對第二個問題,我們可以以在所有輸入的訓練tensor後加上 .type(torch.cuda.FloatTensor),如果是gorund_truth則加上.type(torch.cuda.LongTensor)。這部分也是我們疏忽,我們會盡快更新。