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

Specify a gpu for a network

Browse Source
This commit is contained in:
Shangtong Zhang committed 2018-02-03 10:53:14 -07:00
1 parent fef04b6bf6
commit e91ec3be45
12 files changed
+93 -103
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

component/replay.py
+4 -4
View File
@@ -16,10 +16,10 @@ class Replay:
self.dtype = dtype
self.states = None
self.actions = np.empty(self.memory_size, dtype=np.int8)
self.actions = np.empty(self.memory_size, dtype=np.uint8)
self.rewards = np.empty(self.memory_size)
self.next_states = None
self.terminals = np.empty(self.memory_size, dtype=np.int8)
self.terminals = np.empty(self.memory_size, dtype=np.uint8)
self.pos = 0
self.full = False
@@ -59,10 +59,10 @@ class HybridRewardReplay:
self.dtype = dtype
self.states = None
self.actions = np.empty(self.memory_size, dtype=np.int8)
self.actions = np.empty(self.memory_size, dtype=np.uint8)
self.rewards = None
self.next_states = None
self.terminals = np.empty(self.memory_size, dtype=np.int8)
self.terminals = np.empty(self.memory_size, dtype=np.uint8)
self.pos = 0
self.full = False