From 4d4f3d440bbaefafcb1af61af688783f03e7b7e9 Mon Sep 17 00:00:00 2001 From: Andrew Choi Date: Fri, 13 Mar 2026 11:05:01 -0700 Subject: [PATCH] Add td_error_abs to TDLoss --- alf/algorithms/td_loss.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/alf/algorithms/td_loss.py b/alf/algorithms/td_loss.py index 32269bae8..80721b233 100644 --- a/alf/algorithms/td_loss.py +++ b/alf/algorithms/td_loss.py @@ -20,7 +20,7 @@ import alf from alf.data_structures import LossInfo, namedtuple, StepType from alf.utils.losses import element_wise_squared_loss, iqn_huber_loss -from alf.utils import losses, tensor_utils, value_ops +from alf.utils import tensor_utils, value_ops from alf.utils.summary_utils import safe_mean_hist_summary from alf.utils.normalizers import AdaptiveNormalizer @@ -217,6 +217,8 @@ def _summarize(v, r, td, suffix): safe_mean_hist_summary('values' + suffix, v, mask) safe_mean_hist_summary('returns' + suffix, r, mask) safe_mean_hist_summary("td_error" + suffix, td, mask) + safe_mean_hist_summary('td_error_abs' + suffix, td.abs(), + mask) if value.ndim == 2: _summarize(value, returns, td_error, '')