Skip to content

Commit

Permalink
NN Archive Platform Name (#12)
Browse files Browse the repository at this point in the history
  • Loading branch information
kozlov721 authored Jul 22, 2024
1 parent 09a31fa commit 0e95d50
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion modelconverter/__main__.py
Original file line number Diff line number Diff line change
Expand Up @@ -444,7 +444,7 @@ def convert(
main_stage,
)
generator = ArchiveGenerator(
archive_name=cfg.name,
archive_name=f"{cfg.name}.{target.value.lower()}",
save_path=str(output_path),
cfg_dict=nn_archive.model_dump(),
executables_paths=[
Expand Down

0 comments on commit 0e95d50

Please sign in to comment.