bug fix

PJLab-ADG · Sep 7, 2023 · ef5092a · ef5092a
1 parent d44c395
commit ef5092a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/refining/detzero_refine/models/modules/position_transformer.py b/refining/detzero_refine/models/modules/position_transformer.py
@@ -165,7 +165,7 @@ def get_loss(self, tb_dict=None):
                 self.targets_dict['heading_cls'][i, :num]
             ) / bs
 
-            dir_tmp_loss += reg_loss(
+            dir_tmp_loss = reg_loss(
                 self.preds_dict['heading_reg'][i, :num, :],
                 self.targets_dict['heading_reg'][i, :num, :]
             )