@@ -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