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

Upgrade to PyTorch v0.3

Browse Source
This commit is contained in:
Shangtong Zhang
2017-12-11 23:41:16 -07:00
parent 4697d1d3e9
commit 83a4270bb2
3 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
network/base_network.py
+2 -2
View File
@@ -57,8 +57,8 @@ class ActorCriticNet(BasicNet):
def predict(self, x):
phi = self.forward(x, True)
pre_prob = self.fc_actor(phi)
prob = F.softmax(pre_prob)
log_prob = F.log_softmax(pre_prob)
prob = F.softmax(pre_prob, dim=1)
log_prob = F.log_softmax(pre_prob, dim=1)
value = self.fc_critic(phi)
return prob, log_prob, value