steffen/marl-factory-grid
1
0
Fork 0
mirror of https://github.com/illiumst/marl-factory-grid.git synced 2025-06-18 10:42:53 +02:00
Code Issues Packages Projects Releases Wiki Activity

cleanup algos + adjusted renderer to support "ray casting"

Browse Source
This commit is contained in:
romue
2021-07-27 16:59:24 +02:00
parent 8429e3db9d
commit 9e8d2ac1dc
9 changed files with 102 additions and 254 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
algorithms/m_q_learner.py
View File

@ -25,7 +25,7 @@ class MQLearner(QLearner):
if len(self.buffer) < self.batch_size: return
for _ in range(self.n_grad_steps):
experience = self.buffer.sample(self.batch_size, cer=self.train_every_n_steps)
experience = self.buffer.sample(self.batch_size, cer=self.train_every[-1])
with torch.no_grad():
q_target_next = self.target_q_net(experience.next_observation)