hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-04 17:41:43 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

[RLlib] Tolerate nan metrics in LearnerInfoBuilder. (#27981)

Browse source
This commit is contained in:
Artur Niederfahrenhorst 2022-08-23 19:07:32 +02:00 • committed by GitHub
parent 5fb36d4a7d
commit 56e7800e0b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
rllib/utils/metrics/learner_info.py
View file

@ -106,5 +106,7 @@ def _all_tower_reduce(path, *tower_data):
# Max stats: Reduce max.
elif path[-1].startswith("max_"):
return np.nanmax(tower_data)
if np.isnan(tower_data).all():
return np.nan
# Everything else: Reduce mean.
return np.nanmean(tower_data)