Skip to content

Commit 6eb85cb

Browse files
committed
hotfix(eval): not assert but return if no gt class pt etc.
to avoid killed during eval.
1 parent c0ebc92 commit 6eb85cb

1 file changed

Lines changed: 6 additions & 3 deletions

File tree

src/utils/eval_metric.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -154,9 +154,12 @@ def accumulate_value(
154154
average_range: float,
155155
count: int,
156156
):
157-
assert count > 0, f"count must be greater than 0, got {count}"
158-
assert np.isfinite(average_epe), f"average_epe must be finite, got {average_epe}"
159-
assert np.isfinite(average_range), f"average_range must be finite, got {average_range}"
157+
if count == 0 or np.isnan(average_epe) or np.isnan(average_range):
158+
print("Warning in accumulate_value: count is 0 or average_epe/average_range is NaN, skip this entry.")
159+
return
160+
# assert count > 0, f"count must be greater than 0, got {count}"
161+
# assert np.isfinite(average_epe), f"average_epe must be finite, got {average_epe}"
162+
# assert np.isfinite(average_range), f"average_range must be finite, got {average_range}"
160163

161164
class_idx = self.class_names.index(class_name)
162165
range_bucket_idx = self.range_buckets.index(range_bucket)

0 commit comments

Comments
 (0)