github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

Initialise numpy arrays as float32 in DDPG (#1171)

Browse Source 
float64 is not supported by tinygrad
This commit is contained in:
Stan
2023-07-07 21:05:31 +02:00
committed by GitHub
parent 884b5965de
commit f40f8cd055
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
examples/deep_deterministic_policy_gradient.py
View File

@@ -50,11 +50,11 @@ class Buffer:
self.buffer_counter = 0
self.state_buffer = np.zeros((self.buffer_capacity, num_states))
self.action_buffer = np.zeros((self.buffer_capacity, num_actions))
self.reward_buffer = np.zeros((self.buffer_capacity, 1))
self.next_state_buffer = np.zeros((self.buffer_capacity, num_states))
self.done_buffer = np.zeros((self.buffer_capacity, 1))
self.state_buffer = np.zeros((self.buffer_capacity, num_states), np.float32)
self.action_buffer = np.zeros((self.buffer_capacity, num_actions), np.float32)
self.reward_buffer = np.zeros((self.buffer_capacity, 1), np.float32)
self.next_state_buffer = np.zeros((self.buffer_capacity, num_states), np.float32)
self.done_buffer = np.zeros((self.buffer_capacity, 1), np.float32)
def record(
self, observations: Tuple[Tensor, NDArray, float, NDArray, bool]