diff --git a/tools/eval.py b/tools/eval.py index 2940ca1..5e7d2e6 100644 --- a/tools/eval.py +++ b/tools/eval.py @@ -94,7 +94,9 @@ def main(): model = build_local_model(config, device) model.head.nms = True - + model.head.nms_conf_thre = args.conf + model.head.nms_iou_thre = args.nms + model.cuda(args.local_rank) model.eval()