From e9d8498fe3be8fe8b21036dfe13adb3739be4bee Mon Sep 17 00:00:00 2001 From: Jael Gu Date: Mon, 22 Aug 2022 10:51:46 +0800 Subject: [PATCH] Add distilled weights Signed-off-by: Jael Gu --- nn_fingerprint.py | 2 +- saved_model/{pfann_fma_m.pt => nnfp_fma.pt} | 0 saved_model/nnfp_fma_distill_128.pt | 3 +++ saved_model/pfann_fma_s.pt | 3 --- 4 files changed, 4 insertions(+), 4 deletions(-) rename saved_model/{pfann_fma_m.pt => nnfp_fma.pt} (100%) create mode 100644 saved_model/nnfp_fma_distill_128.pt delete mode 100644 saved_model/pfann_fma_s.pt diff --git a/nn_fingerprint.py b/nn_fingerprint.py index 340b5d2..112ef8c 100644 --- a/nn_fingerprint.py +++ b/nn_fingerprint.py @@ -60,7 +60,7 @@ class NNFingerprint(NNOperator): log.info('Loading model...') if model_path is None: path = str(Path(__file__).parent) - model_path = os.path.join(path, 'saved_model', 'pfann_fma_m.pt') + model_path = os.path.join(path, 'saved_model', 'nnfp_fma.pt') if model_path.endswith('.onnx'): log.warning('Using onnx.') self.model = onnxruntime.InferenceSession( diff --git a/saved_model/pfann_fma_m.pt b/saved_model/nnfp_fma.pt similarity index 100% rename from saved_model/pfann_fma_m.pt rename to saved_model/nnfp_fma.pt diff --git a/saved_model/nnfp_fma_distill_128.pt b/saved_model/nnfp_fma_distill_128.pt new file mode 100644 index 0000000..89727d3 --- /dev/null +++ b/saved_model/nnfp_fma_distill_128.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4bba609dd72ea45fb14b9470d73dbbbb02a3d484eb4e7de468c4bf81c29c7de5 +size 24820057 diff --git a/saved_model/pfann_fma_s.pt b/saved_model/pfann_fma_s.pt deleted file mode 100644 index b9ab647..0000000 --- a/saved_model/pfann_fma_s.pt +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:470214d98a67b2d9d54d3f73ea389ae922dd41a10e2f1578e0d3014974996ec0 -size 67677593