Fixed minor bug in sac post load

MushroomRL · Jan 30, 2024 · 47a3fce · 47a3fce
1 parent 146ba8e
commit 47a3fce
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mushroom_rl/algorithms/actor_critic/deep_actor_critic/sac.py b/mushroom_rl/algorithms/actor_critic/deep_actor_critic/sac.py
@@ -319,7 +319,7 @@ def _post_load(self):
         self._update_alpha_optimizer_parameters()
 
     def _update_alpha_optimizer_parameters(self):
-        if self._optimizer is not None:
+        if self._alpha_optim is not None:
             TorchUtils.update_optimizer_parameters(self._alpha_optim, [self._log_alpha])
 
     @property