Skip to content

Commit 53bf20d

Browse files
authored
Update train_net.py
1 parent 7ee80e8 commit 53bf20d

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

train_net.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -476,7 +476,7 @@ def test_save_features(cls, cfg, model, evaluators=None):
476476
print(idx)
477477

478478
@classmethod
479-
def test_visual_openset(cls, cfg, model, eval_visual_openset_combine, evaluators=None):
479+
def test_visual_openset(cls, cfg, model, evaluators=None):
480480
# build dataloade
481481
dataloaders = cls.build_test_loader(cfg, dataset_name=None)
482482
dataset_names = cfg['DATASETS']['TEST']
@@ -613,7 +613,7 @@ def main(args=None):
613613
cfg.MODEL.WEIGHTS, resume=args.resume
614614
)
615615
if args.eval_visual_openset:
616-
res = Trainer.test_visual_openset(cfg, model, args.eval_visual_openset_combine)
616+
res = Trainer.test_visual_openset(cfg, model)
617617
elif args.eval_track_prev:
618618
res = Trainer.test_tracking_prev(cfg, model)
619619
elif args.eval_get_content_features:

0 commit comments

Comments
 (0)