wassname/DeepRL
1
0
Fork 0
mirror of https://github.com/wassname/DeepRL.git synced 2026-09-09 11:13:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add keepdim

Browse Source
This commit is contained in:
Shangtong Zhang
2018-03-31 21:46:16 -06:00
parent 0eaba783e8
commit e7e94d715b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
network/base_network.py
+1 -1
View File
@@ -88,7 +88,7 @@ class DuelingNet(BasicNet):
phi = self.forward(x)
value = self.fc_value(phi)
advantange = self.fc_advantage(phi)
q = value.expand_as(advantange) + (advantange - advantange.mean(1).expand_as(advantange))
q = value.expand_as(advantange) + (advantange - advantange.mean(1, keepdim=True).expand_as(advantange))
if to_numpy:
return q.cpu().data.numpy()
return q