update

CodSpeedHQ · Sep 5, 2024 · e36afbf · e36afbf
1 parent 37699e4
commit e36afbf
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/kornia/contrib/models/rt_detr/architecture/rtdetr_head.py b/kornia/contrib/models/rt_detr/architecture/rtdetr_head.py
@@ -276,7 +276,8 @@ def __init__(
         super().__init__()
         self.num_queries = num_queries
         # TODO: verify this is correct
-        assert len(in_channels) <= num_levels
+        if len(in_channels) > num_levels:
+            raise ValueError(f"`num_levels` cannot be greater than {len(in_channels)}. Got {num_levels}.")
         self.num_levels = num_levels
 
         # build the input projection layers