質問編集履歴
3
補足
test
CHANGED
File without changes
|
test
CHANGED
@@ -18,6 +18,8 @@
|
|
18
18
|
|
19
19
|
```
|
20
20
|
|
21
|
+
このような文と共にさいごRuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu)と表示されてしまいます。
|
22
|
+
|
21
23
|
Epoch gpu_mem box obj cls total labels img_size
|
22
24
|
0% 0/1 [00:05<?, ?it/s]
|
23
25
|
Traceback (most recent call last):
|
@@ -29,6 +31,7 @@
|
|
29
31
|
bs_aux, as_aux_, gjs_aux, gis_aux, targets_aux, anchors_aux = self.build_targets2(p[:self.nl], targets, imgs)
|
30
32
|
File "/content/drive/MyDrive/Colab Notebooks/yolo/yolov7/utils/loss.py", line 1559, in build_targets2
|
31
33
|
from_which_layer = from_which_layer[fg_mask_inboxes]
|
34
|
+
|
32
35
|
RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu)
|
33
36
|
|
34
37
|
|
2
タイトル
test
CHANGED
@@ -1 +1 @@
|
|
1
|
-
RuntimeError
|
1
|
+
YOLO v7 とVOTTを利用した物体検出でRuntimeErrorが出てしまう
|
test
CHANGED
File without changes
|
1
改行
test
CHANGED
File without changes
|
test
CHANGED
@@ -142,6 +142,8 @@
|
|
142
142
|
#変更終了
|
143
143
|
|
144
144
|
from typing_extensions import dataclass_transform
|
145
|
+
|
146
|
+
|
145
147
|
!python train_aux.py --workers 2 --batch-size 8 \
|
146
148
|
--data /content/drive/MyDrive/Colab Notebooks/yolo/yolov7/data/data.yaml \
|
147
149
|
--cfg cfg/training/yolov7-e6.yaml \
|