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