Fixed error on clip_range None

2024-01-23 09:22:58 +01:00
parent e3f4c511bf
commit 252710489e
1 changed files with 2 additions and 1 deletions
@@ -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)):