saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-06-05 00:00:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add sanity check for clip_range_vf

Browse source
This commit is contained in:
Antonin RAFFIN 2020-04-23 09:59:09 +02:00
parent 02b452a9f9
commit 271a0a7818
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
torchy_baselines/ppo/ppo.py
View file

@ -135,6 +135,10 @@ class PPO(BaseRLModel):
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')
self.clip_range_vf = get_schedule_fn(self.clip_range_vf)
def collect_rollouts(self,