remove dim assert

Visual-Behavior · Feb 16, 2023 · fbd9169 · fbd9169
1 parent 49ea8b5
commit fbd9169
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/alonet/detr/misc.py b/alonet/detr/misc.py
@@ -14,7 +14,6 @@ def wrapper(instance, frames: Union[torch.Tensor, Frame], is_export_onnx=False,
             # because torch.onnx.export accepts only torch.Tensor or None
             if hasattr(instance, "tracing") and instance.tracing:
                 assert isinstance(frames, torch.Tensor)
-                assert frames.shape[1] == 4  # rgb 3 + mask 1
                 kwargs["is_tracing"] = None
                 if is_export_onnx is None:
                     return forward(instance, frames, is_export_onnx=None, **kwargs)