mirror of
https://github.com/wassname/Run-Skeleton-Run.git
synced 2026-06-27 16:13:51 +08:00
init policy_net
FYI: I think this is a typo
This commit is contained in:
+1
-1
@@ -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)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user