diff options
author | Jan Buethe <jbuethe@amazon.de> | 2023-11-29 21:53:52 +0300 |
---|---|---|
committer | Jean-Marc Valin <jmvalin@amazon.com> | 2023-12-01 00:23:20 +0300 |
commit | 2f94bdf83aaf8eb3a9d86eb0faae8d8e9ca2ddd0 (patch) | |
tree | 6d59026659832debf92859acd44c071ab64c13ce | |
parent | f8c5b54d48a011dcbe5ce186118dba4df69760f8 (diff) |
bugfix
-rw-r--r-- | dnn/torch/osce/export_model_weights.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/dnn/torch/osce/export_model_weights.py b/dnn/torch/osce/export_model_weights.py index 05fd0265..03d45212 100644 --- a/dnn/torch/osce/export_model_weights.py +++ b/dnn/torch/osce/export_model_weights.py @@ -112,7 +112,7 @@ if __name__ == "__main__": ''') for i, s in enumerate(model.numbits_embedding.scale_factors): - cwriter.header.write(f"#define LACE_NUMBITS_SCALE_{i} {float(s.detach().cpu())}\n") + cwriter.header.write(f"#define {model_name.upper()}_NUMBITS_SCALE_{i} {float(s.detach().cpu())}\n") # dump layers osce_dump_generic(cwriter, model_name, model) |