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

Support shared stats for continuous A3C

Browse Source
This commit is contained in:
Shangtong Zhang
2017-10-04 20:58:42 -06:00
parent 3935dfc428
commit c773da7d08
7 changed files with 105 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files
network/network.py
+1 -1
View File
@@ -51,7 +51,7 @@ class VanillaNet(BasicNet):
# Base class for actor critic method
class ActorCriticNet(BasicNet):
def predict(self, x, _):
def predict(self, x):
phi = self.forward(x, True)
pre_prob = self.fc_actor(phi)
prob = F.softmax(pre_prob)