回答編集履歴
2
修正
answer
CHANGED
|
@@ -5,7 +5,7 @@
|
|
|
5
5
|
|
|
6
6
|
> with torch.no_grad():をValidate関数内に以下のように付け足した。
|
|
7
7
|
|
|
8
|
-
推論時はバックプロパゲーションをしないので、中間層の出力を記録しておく必要がないので、これ有効です。
|
|
8
|
+
推論時はバックプロパゲーションをしないので、中間層の出力を記録しておく必要がないので、これは有効です。
|
|
9
9
|
|
|
10
10
|
> 1epoch回すだけで1時間学習にかかるのでこのエラーは正直つらいです。
|
|
11
11
|
|
1
修正
answer
CHANGED
|
@@ -5,4 +5,8 @@
|
|
|
5
5
|
|
|
6
6
|
> with torch.no_grad():をValidate関数内に以下のように付け足した。
|
|
7
7
|
|
|
8
|
-
推論時はバックプロパゲーションをしないので、中間層の出力を記録しておく必要がないので、これ有効です。
|
|
8
|
+
推論時はバックプロパゲーションをしないので、中間層の出力を記録しておく必要がないので、これ有効です。
|
|
9
|
+
|
|
10
|
+
> 1epoch回すだけで1時間学習にかかるのでこのエラーは正直つらいです。
|
|
11
|
+
|
|
12
|
+
学習のほうを1イテレーションだけ回したあとにすぐ break してバリデーションすれば、今の設定でメモリ不足かどうかはすぐに確認できると思います。
|