We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent fbe9fd3 commit 6984144Copy full SHA for 6984144
UtilsRL/rl/buffer/prioritized_replay.py
@@ -93,7 +93,7 @@ def batch_update(self, batch_idx, metric_value):
93
if metric_value.shape == ():
94
metric_value = np.asarray([metric_value, ])
95
# update crendential
96
- self.max_metric_value = max(metric_value, self.max_metric_value)
+ self.max_metric_value = max(np.max(metric_value), self.max_metric_value)
97
self.sum_tree.update(batch_idx, self.metric_fn(metric_value))
98
self.min_tree.update(batch_idx, self.metric_fn(metric_value))
99
0 commit comments