From bc89bb2eed64df4a99c54fb7041cf63cef0d595f Mon Sep 17 00:00:00 2001 From: Shangtong Zhang Date: Fri, 27 Apr 2018 16:31:33 -0600 Subject: [PATCH] Init FCBody --- network/network_bodies.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/network/network_bodies.py b/network/network_bodies.py index fbac4bd..ef60fc9 100644 --- a/network/network_bodies.py +++ b/network/network_bodies.py @@ -27,7 +27,7 @@ class FCBody(nn.Module): def __init__(self, state_dim, hidden_units=(64, 64), gate=F.tanh): super(FCBody, self).__init__() dims = (state_dim, ) + hidden_units - self.layers = nn.ModuleList([nn.Linear(dim_in, dim_out) for dim_in, dim_out in zip(dims[:-1], dims[1:])]) + self.layers = nn.ModuleList([layer_init(nn.Linear(dim_in, dim_out)) for dim_in, dim_out in zip(dims[:-1], dims[1:])]) self.gate = gate self.feature_dim = dims[-1]