Skip to content

Commit c1fad3b

Browse files
committed
fix training bug
1 parent 3c2caff commit c1fad3b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

cosypose/training/train_pose.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,6 @@ def load_model(run_id):
106106
base_pred_kwargs = dict(
107107
pose_predictor=predictor,
108108
mv_predictor=None,
109-
use_gt_detections=False,
110109
skip_mv=True,
111110
)
112111
for ds_name in args.test_ds_names:
@@ -198,6 +197,7 @@ def train_pose(args):
198197
args.train_refiner = args.TCO_input_generator == 'gt+noise'
199198
args.train_coarse = not args.train_refiner
200199
args.save_dir = EXP_DIR / args.run_id
200+
args = check_update_config(args)
201201

202202
logger.info(f"{'-'*80}")
203203
for k, v in args.__dict__.items():

0 commit comments

Comments
 (0)