diff --git a/rl/agent/double_dqn.py b/rl/agent/double_dqn.py index 9a6cf45..92e104d 100644 --- a/rl/agent/double_dqn.py +++ b/rl/agent/double_dqn.py @@ -45,9 +45,9 @@ def recompile_model(self, sys_vars): '''rotate and recompile both models''' if self.epi_change_lr is not None: self.switch_models() # to model_2 - self.recompile_model(sys_vars) + super(DoubleDQN, self).recompile_model(sys_vars) self.switch_models() # back to model - self.recompile_model(sys_vars) + super(DoubleDQN, self).recompile_model(sys_vars) return self.model def compute_Q_states(self, minibatch):