From b5a67b062f600d7c9e0b814284402994d040fa6f Mon Sep 17 00:00:00 2001 From: Mike Clark Date: Mon, 22 Jan 2018 17:40:35 +0800 Subject: [PATCH] init policy_net FYI: I think this is a typo --- ddpg/nets.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ddpg/nets.py b/ddpg/nets.py index 06b2807..b80e472 100644 --- a/ddpg/nets.py +++ b/ddpg/nets.py @@ -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)