yjh0410 há 1 ano atrás
pai
commit
8a08459bf3
1 ficheiros alterados com 1 adições e 1 exclusões
  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