回答編集履歴
2
test
CHANGED
@@ -7,7 +7,8 @@
|
|
7
7
|
検討ください。
|
8
8
|
|
9
9
|
---
|
10
|
+
ソースコードに以下の文面がありました。
|
10
|
-
|
11
|
+
RTXはもっとバケモノみたいなメモリを積んでいそうですが、設定したバッチサイズとGPUのメモリはいくつだったのでしょうか?
|
11
12
|
> 1) ResNet50 (v2) backbone.
|
12
13
|
> Train with 6 layers of feature maps.
|
13
14
|
> Pls adjust batch size depending on your GPU memory.
|
1
test
CHANGED
@@ -5,3 +5,12 @@
|
|
5
5
|
|
6
6
|
やっていないので保証はできませんが、filter=256を128や64にすれば劇的にメモリが減るはずです。
|
7
7
|
検討ください。
|
8
|
+
|
9
|
+
---
|
10
|
+
ソースコードに以下の文面がありました。RTXはもっとバケモノみたいなメモリを積んでいそうですが、バッチサイズはいくつだったのでしょうか?
|
11
|
+
> 1) ResNet50 (v2) backbone.
|
12
|
+
> Train with 6 layers of feature maps.
|
13
|
+
> Pls adjust batch size depending on your GPU memory.
|
14
|
+
> For 1060 with 6GB, --batch-size=1. For V100 with 32GB,
|
15
|
+
> --batch-size=4
|
16
|
+
|