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)。這部分也是我們疏忽，我們會盡快更新。