Update train_net.py

UX-Decoder · Apr 8, 2024 · 53bf20d · 53bf20d
1 parent 7ee80e8
commit 53bf20d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/train_net.py b/train_net.py
@@ -476,7 +476,7 @@ def test_save_features(cls, cfg, model, evaluators=None):
  print(idx)
 
  @classmethod
- def test_visual_openset(cls, cfg, model, eval_visual_openset_combine, evaluators=None):
+ def test_visual_openset(cls, cfg, model, evaluators=None):
  # build dataloade
  dataloaders = cls.build_test_loader(cfg, dataset_name=None)
  dataset_names = cfg['DATASETS']['TEST']
@@ -613,7 +613,7 @@ def main(args=None):
  cfg.MODEL.WEIGHTS, resume=args.resume
  )
  if args.eval_visual_openset:
- res = Trainer.test_visual_openset(cfg, model, args.eval_visual_openset_combine)
+ res = Trainer.test_visual_openset(cfg, model)
  elif args.eval_track_prev:
  res = Trainer.test_tracking_prev(cfg, model)
  elif args.eval_get_content_features: