WIP

waifairer · Oct 20, 2023 · d5ece13 · d5ece13
1 parent 4d83a1c
commit d5ece13
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/mask_rcnn/loss.py b/examples/mask_rcnn/loss.py
@@ -208,7 +208,7 @@ def concat_box_prediction_layers(box_cls, box_regression):
   # being concatenated as well)
   if DEBUG > 0:
     print_gpu_memory("prior to cat")
-  box_cls = Tensor.cat(*box_cls_flattened, dim=1).reshape(-1, C)
+  box_cls = Tensor.cat(*box_cls_flattened, dim=1)#.reshape(-1, C)
   if DEBUG > 0:
     print_gpu_memory("after boxcls cat")
     box_cls.realize()