This commit is contained in:
wassname
2018-01-18 17:23:35 +08:00
parent d9b67e5f9a
commit a87a3ad7bb
3 changed files with 14 additions and 158 deletions
-1
View File
@@ -150,7 +150,6 @@ class CriticHead(nn.Module):
def forward(self, observation):
x = self.base.forward(observation)
# x = torch.cat((x, action), dim=1)
x = self.value_net.forward(x)
return x