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

Log wall time

Browse Source
This commit is contained in:
Shangtong Zhang
2017-10-14 16:38:51 -06:00
parent c04b2a395f
commit 3ed6f2a9db
4 changed files with 70 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
network/network.py
+3
View File
@@ -19,6 +19,9 @@ class BasicNet:
self.LSTM = LSTM
if self.gpu:
self.cuda()
self.FloatTensor = torch.cuda.FloatTensor
else:
self.FloatTensor = torch.FloatTensor
def to_torch_variable(self, x, dtype='float32'):
if isinstance(x, Variable):