yolov3spp コード実行プロセス中に train.py ファイルを使用すると失敗する問題を解決する

 

RuntimeError: インデックスは CPU 上、またはインデックス付きテンソルと同じデバイス上にある必要があります (CPU

この問題を報告しました

util.py スクリプトに追加します

at = at.to(j.device)

これら 2 つの変数を同じデバイス上で train.py ファイルに直接実行するようにします。