Skip to content

Commit

Permalink
update qat_export.py
Browse files Browse the repository at this point in the history
  • Loading branch information
liqingyuan02 committed Sep 22, 2022
1 parent 1a4afb7 commit 0753cf4
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions tools/qat/qat_export.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,6 +38,7 @@
parser.add_argument('--img-size', nargs='+', type=int, default=[640, 640], help='image size') # height, width
parser.add_argument('--conf', type=str, default='../../configs/repopt/yolov6s_opt_qat.py', help='model config')
parser.add_argument('--export-batch-size', type=int, default=None, help='export batch size')
parser.add_argument('--calib', action='store_true', default=False, help='calibrated model')
parser.add_argument('--fuse-bn', action='store_true', help='fuse bn')
parser.add_argument('--graph-opt', action='store_true', help='enable graph optimizer')
parser.add_argument('--inplace', action='store_true', help='set Detect() inplace=True')
Expand Down

0 comments on commit 0753cf4

Please sign in to comment.