diff options
| author | Jan Buethe <jbuethe@amazon.de> | 2023-12-18 11:31:52 +0300 |
|---|---|---|
| committer | Jean-Marc Valin <jmvalin@amazon.com> | 2023-12-20 07:01:33 +0300 |
| commit | 282382764805f0999776bf381f416d7b749d1633 (patch) | |
| tree | e21625195f899915bfc25724b161bcac3c9e193d | |
| parent | ae706b9fb726dfeec40796566d3ceee155a82e96 (diff) | |
added missing f in export_model_weights.py
| -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 786d3200..da90a39a 100644 --- a/dnn/torch/osce/export_model_weights.py +++ b/dnn/torch/osce/export_model_weights.py @@ -159,7 +159,7 @@ if __name__ == "__main__": ''') for i, s in enumerate(model.numbits_embedding.scale_factors): - cwriter.header.write(f"#define {model_name.upper()}_NUMBITS_SCALE_{i} {float(s.detach().cpu())}\n") + cwriter.header.write(f"#define {model_name.upper()}_NUMBITS_SCALE_{i} {float(s.detach().cpu())}f\n") # dump layers if model_name in schedules and args.quantize: |