質問編集履歴
1
記載ミス
test
CHANGED
|
File without changes
|
test
CHANGED
|
@@ -11,8 +11,6 @@
|
|
|
11
11
|
loss = criterion(output,targets[i])
|
|
12
12
|
|
|
13
13
|
loss=loss*weights[i]
|
|
14
|
-
|
|
15
|
-
print(loss.is_cuda)
|
|
16
14
|
|
|
17
15
|
with amp.scale_loss(loss, optimizer) as scaled_loss:
|
|
18
16
|
|