dodox/metastable-baselines2
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fixed error on clip_range None

Browse Source
This commit is contained in:
Dominik Moritz Roth 2024-01-23 09:22:58 +01:00
parent e3f4c511bf
commit 252710489e
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
metastable_baselines2/trpl/trpl.py
View File

@ -186,7 +186,8 @@ class TRPL(BetterOnPolicyAlgorithm):
super()._setup_model()
# Initialize schedules for policy/value clipping
self.clip_range = get_schedule_fn(self.clip_range)
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)):
assert self.clip_range_vf > 0, "`clip_range_vf` must be positive, " "pass `None` to deactivate vf clipping"