diff --git a/docs/misc/changelog.rst b/docs/misc/changelog.rst index 8df321129..78eb2bd0e 100644 --- a/docs/misc/changelog.rst +++ b/docs/misc/changelog.rst @@ -3,7 +3,7 @@ Changelog ========== -Release 2.4.0a4 (WIP) +Release 2.4.0a5 (WIP) -------------------------- Breaking Changes: @@ -20,6 +20,7 @@ Bug Fixes: - ``CallbackList`` now sets the ``.parent`` attribute of child callbacks to its own ``.parent``. (will-maclean) - Fixed error when loading a model that has ``net_arch`` manually set to ``None`` (@jak3122) - Set requirement numpy<2.0 until PyTorch is compatible (https://github.com/pytorch/pytorch/issues/107302) +- Updated DQN optimizer input to only include q_network parameters, removing the target_q_network ones (@corentinlger) `SB3-Contrib`_ ^^^^^^^^^^^^^^ diff --git a/stable_baselines3/dqn/policies.py b/stable_baselines3/dqn/policies.py index 9d2cf94df..bfefc8137 100644 --- a/stable_baselines3/dqn/policies.py +++ b/stable_baselines3/dqn/policies.py @@ -167,7 +167,7 @@ def _build(self, lr_schedule: Schedule) -> None: # Setup optimizer with initial learning rate self.optimizer = self.optimizer_class( # type: ignore[call-arg] - self.parameters(), + self.q_net.parameters(), lr=lr_schedule(1), **self.optimizer_kwargs, ) diff --git a/stable_baselines3/version.txt b/stable_baselines3/version.txt index 2d22b1587..a1fd35b5f 100644 --- a/stable_baselines3/version.txt +++ b/stable_baselines3/version.txt @@ -1 +1 @@ -2.4.0a4 +2.4.0a5