回答編集履歴
1
改善方法分かりました。
test
CHANGED
@@ -3,3 +3,15 @@
|
|
3
3
|
のところを実行すると、
|
4
4
|
NameError: name 'device' is not defined
|
5
5
|
と出てしまいますが・・・。
|
6
|
+
|
7
|
+
調べたところ
|
8
|
+
|
9
|
+
# GPUの利用
|
10
|
+
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
11
|
+
net = net.to(device)
|
12
|
+
|
13
|
+
という風に、netの上にdeviceを加えた所、ここは通りました。
|
14
|
+
訂正も必要みたいですね。
|
15
|
+
|
16
|
+
|
17
|
+
ところで、この本のコードでも画像分類はできるみたいですがこれと、質問のコード、どっちが高性能なんでしょうか?
|