init policy_net

FYI: I think this is a typo
This commit is contained in:
Mike Clark
2018-01-22 17:40:35 +08:00
committed by GitHub
parent a55757c984
commit b5a67b062f
+1 -1
View File
@@ -43,7 +43,7 @@ class Actor(nn.Module):
if isinstance(layer, nn.Linear):
layer.weight.data = fanin_init(layer.weight.data.size())
for layer in self.feature_net.net:
for layer in self.policy_net.net:
if isinstance(layer, nn.Linear):
layer.weight.data.uniform_(-init_w, init_w)