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

Specify a gpu for a network

Browse Source
This commit is contained in:
Shangtong Zhang
2018-02-03 10:53:14 -07:00
parent fef04b6bf6
commit e91ec3be45
12 changed files with 93 additions and 103 deletions
Download Patch File Download Diff File Expand all files Collapse all files
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