Skip to content

Commit 9c38bcc

Browse files
committed
hotfix(eval): not assert but return if no gt class pt etc.
to avoid killed during eval.
1 parent 7fb4973 commit 9c38bcc

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

src/utils/eval_metric.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -147,9 +147,12 @@ def accumulate_value(
147147
average_range: float,
148148
count: int,
149149
):
150-
assert count > 0, f"count must be greater than 0, got {count}"
151-
assert np.isfinite(average_epe), f"average_epe must be finite, got {average_epe}"
152-
assert np.isfinite(average_range), f"average_range must be finite, got {average_range}"
150+
if count == 0 or np.isnan(average_epe) or np.isnan(average_range):
151+
print("Warning in accumulate_value: count is 0 or average_epe/average_range is NaN, skip this entry.")
152+
return
153+
# assert count > 0, f"count must be greater than 0, got {count}"
154+
# assert np.isfinite(average_epe), f"average_epe must be finite, got {average_epe}"
155+
# assert np.isfinite(average_range), f"average_range must be finite, got {average_range}"
153156

154157
class_idx = self.class_names.index(class_name)
155158
range_bucket_idx = self.range_buckets.index(range_bucket)

0 commit comments

Comments
 (0)