logo
Loading...

Build SSD GPU 訓練問題 - Cupoy

在執行 Day 29 Build SSD_VGG.ipynb 時遇到兩個問題,在基本設定的程式碼中原...

cvdl,cvdl-d29

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