minor fix

Wuziyi616 · Feb 19, 2023 · d874cb2 · d874cb2
1 parent 27e31de
commit d874cb2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/multi_part_assembly/models/modules/base_model.py b/multi_part_assembly/models/modules/base_model.py
@@ -261,7 +261,7 @@ def _calc_loss(self, out_dict, data_dict):
         trans_loss = trans_l2_loss(pred_trans, new_trans, valids)
         rot_pt_cd_loss = rot_points_cd_loss(part_pcs, pred_rot, new_rot,
                                             valids)
-        transform_pt_cd_loss, gt_trans_pts, pred_trans_pts = shape_cd_loss(
+        transform_pt_cd_loss, pred_trans_pts, gt_trans_pts = shape_cd_loss(
             part_pcs,
             pred_trans,
             new_trans,