Fix error in build.py

JacobChalk · Aug 2, 2024 · 7b69d3a · 7b69d3a
1 parent b8e359e
commit 7b69d3a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/detection/time_interval_machine/models/build.py b/detection/time_interval_machine/models/build.py
@@ -33,6 +33,7 @@ def build_model(args, gpu_id=None):
  data_modality=args.data_modality,
  num_feats=args.num_feats,
  include_verb_noun=args.include_verb_noun,
+ iou_threshold=args.iou_threshold,
  label_smoothing=args.label_smoothing
  )
 
@@ -62,4 +63,4 @@ def build_model(args, gpu_id=None):
  find_unused_parameters=False
  )
 
- return model, args
+ return model, args