Fixed error on clip_range None

2024-01-23 09:22:58 +01:00 · 2024-01-23 09:22:58 +01:00 · 252710489e
commit 252710489e
parent e3f4c511bf
1 changed files with 2 additions and 1 deletions
--- a/metastable_baselines2/trpl/trpl.py
+++ b/metastable_baselines2/trpl/trpl.py
@ -186,6 +186,7 @@ class TRPL(BetterOnPolicyAlgorithm):
        super()._setup_model()

        # Initialize schedules for policy/value clipping
+        if self.clip_range is not None:
            self.clip_range = get_schedule_fn(self.clip_range)
        if self.clip_range_vf is not None:
            if isinstance(self.clip_range_vf, (float, int)):