質問編集履歴
2
タイトルの改善
title
CHANGED
|
@@ -1,1 +1,1 @@
|
|
|
1
|
-
pfrl,pytorchでのR2D2の実装
|
|
1
|
+
pfrl,pytorchでのR2D2の実装の際の引数のエラー
|
body
CHANGED
|
File without changes
|
1
誤字修正
title
CHANGED
|
File without changes
|
body
CHANGED
|
@@ -84,10 +84,10 @@
|
|
|
84
84
|
#nn.init.kaiming_normal_(self.al1.weight)
|
|
85
85
|
|
|
86
86
|
self.fc2 = nn.Linear(512, 512)
|
|
87
|
-
nn.init.kaiming_normal_(self.
|
|
87
|
+
nn.init.kaiming_normal_(self.fc2.weight)
|
|
88
88
|
|
|
89
89
|
self.fc3 = nn.Linear(512, n_actions*n_atoms)
|
|
90
|
-
nn.init.kaiming_normal_(self.
|
|
90
|
+
nn.init.kaiming_normal_(self.fc3.weight)
|
|
91
91
|
|
|
92
92
|
def forward(self, state):
|
|
93
93
|
img = state[:,:-self.n_added_input]
|