mirror of
https://github.com/wassname/DeepRL.git
synced 2026-10-29 12:40:58 +08:00
Fix a minor bug of option critic
This commit is contained in:
1 parent
7066fb8e89
commit
d7c537707a
1 file changed
+1
-1
@@ -52,7 +52,7 @@ class OptionCriticAgent(BaseAgent):
|
||||
options_next = np.copy(options)
|
||||
dice = np.random.rand(len(options_next))
|
||||
for j in range(len(dice)):
|
||||
if dice[j] < np_betas_next[j]:
|
||||
if dice[j] < np_betas_next[j] or terminals[j]:
|
||||
options_next[j] = self.policy.sample(np_q_options_next[j])
|
||||
for i, terminal in enumerate(terminals):
|
||||
if terminals[i]:
|
||||
|
||||
Reference in new issue
Block a user