Add export args to model metadata (#18472)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
parent
990424cb63
commit
1f1f355a9e
1 changed files with 3 additions and 1 deletions
|
|
@ -250,7 +250,8 @@ class Exporter:
|
|||
self.device = select_device("cpu" if self.args.device is None else self.args.device)
|
||||
|
||||
# Argument compatibility checks
|
||||
validate_args(fmt, self.args, fmts_dict["Arguments"][flags.index(True) + 1])
|
||||
fmt_keys = fmts_dict["Arguments"][flags.index(True) + 1]
|
||||
validate_args(fmt, self.args, fmt_keys)
|
||||
if imx and not self.args.int8:
|
||||
LOGGER.warning("WARNING ⚠️ IMX only supports int8 export, setting int8=True.")
|
||||
self.args.int8 = True
|
||||
|
|
@ -369,6 +370,7 @@ class Exporter:
|
|||
"batch": self.args.batch,
|
||||
"imgsz": self.imgsz,
|
||||
"names": model.names,
|
||||
"args": {k: v for k, v in self.args if k in fmt_keys},
|
||||
} # model metadata
|
||||
if model.task == "pose":
|
||||
self.metadata["kpt_shape"] = model.model[-1].kpt_shape
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue