From ef5092ac9d0457f60306bb69fb5add19d2fb6dde Mon Sep 17 00:00:00 2001 From: superkoma <269492259@qq.com> Date: Thu, 7 Sep 2023 18:19:54 +0800 Subject: [PATCH] bug fix --- refining/detzero_refine/models/modules/position_transformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/refining/detzero_refine/models/modules/position_transformer.py b/refining/detzero_refine/models/modules/position_transformer.py index 3eb4b1f..4f0ad53 100644 --- 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, :] )