yjh0410 hace 1 año
padre
commit
8a08459bf3
Se han modificado 1 ficheros con 1 adiciones y 1 borrados
  1. 1 1
      yolo/benchmark.py

+ 1 - 1
yolo/benchmark.py

@@ -99,7 +99,7 @@ if __name__ == '__main__':
     model = build_model(args, cfg, is_val=False)
 
     # Load trained weight
-    model = load_weight(model, args.weight, args.fuse_conv_bn)
+    model = load_weight(model, args.weight, args.fuse_conv_bn, rep_conv=True)
     model.to(device).eval()
         
     # Run