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 A2C

Browse Source
This commit is contained in:
Shangtong Zhang
2018-01-31 16:19:33 -07:00
parent 63c5d5ea54
commit ceaf83bca3
7 changed files with 201 additions and 123 deletions
Download Patch File Download Diff File Expand all files Collapse all files
network/base_network.py
+2
View File
@@ -18,8 +18,10 @@ class BasicNet:
if self.gpu:
self.cuda()
self.FloatTensor = torch.cuda.FloatTensor
self.LongTensor = torch.cuda.LongTensor
else:
self.FloatTensor = torch.FloatTensor
self.LongTensor = torch.LongTensor
def to_torch_variable(self, x, dtype='float32'):
if isinstance(x, Variable):