1. 程式人生 > >RefineDet -pytorch程式碼記錄

RefineDet -pytorch程式碼記錄

1、RuntimeError: copy_if failed to synchronize: device-side assert triggered

百度搜索說是標籤要從0到N-1;N是類別數

 很奇怪原本沒有-1,輸出label_idx就是從0開始的,

2、expected 0 got 512

這個加在預訓練的VGG16報錯了,也是根據github上下載的,待後續看看怎麼解決