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

Another issue with clip_range=None

Browse Source
This commit is contained in:
Dominik Moritz Roth 2024-01-23 18:32:22 +01:00
parent 8bbb01504b
commit 53505a80ad
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

@ -330,7 +330,8 @@ class TRPL(BetterOnPolicyAlgorithm):
self.logger.record("train/std", th.exp(self.policy.log_std).mean().item())
self.logger.record("train/n_updates", self._n_updates, exclude="tensorboard")
self.logger.record("train/clip_range", clip_range)
if self.clip_range is not None:
self.logger.record("train/clip_range", clip_range)
if self.clip_range_vf is not None:
self.logger.record("train/clip_range_vf", clip_range_vf)