diff --git a/README.md b/README.md index 161f89b..f0be45d 100644 --- a/README.md +++ b/README.md @@ -128,7 +128,7 @@ shape_ranges=dict( ) trt_model = mmdet2trt(cfg_path, weight_path, - shape_ranges=shape_ranges, + shape_ranges=shape_ranges, fp16_mode=True) # save converted model