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

Refactor DDPG

Browse Source
This commit is contained in:
Shangtong Zhang committed 2018-04-25 23:38:17 -06:00
1 parent ceea0a474b
commit 2eefb04b9d
2 files changed
+4 -7
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

network/network_utils.py
+2
View File
@@ -18,6 +18,8 @@ class BaseNet:
self.to(self.device)
def tensor(self, x):
if isinstance(x, torch.Tensor):
return x
x = torch.tensor(x, device=self.device, dtype=torch.float32)
return x