diff options
author | Jean-Marc Valin <jmvalin@amazon.com> | 2023-11-29 10:50:24 +0300 |
---|---|---|
committer | Jean-Marc Valin <jmvalin@amazon.com> | 2023-11-29 10:50:24 +0300 |
commit | 0d823c137c5d733a5240ba2055bc38bfd29398d1 (patch) | |
tree | e05f639e371993618e66a656371bc4bb2304ddb9 | |
parent | 443510c2d12e8fff311111e9bfbfe5434ead8616 (diff) |
Add a script to shrink the DNN models
Removes float debug weights, as well as useless spaces
-rwxr-xr-x | scripts/shrink_model.sh | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/scripts/shrink_model.sh b/scripts/shrink_model.sh new file mode 100755 index 00000000..ae4424ae --- /dev/null +++ b/scripts/shrink_model.sh @@ -0,0 +1,13 @@ +#!/bin/sh + +for i in fargan_data.c pitchdnn_data.c dred_rdovae_dec_data.c dred_rdovae_enc_data.c +do + cat dnn/$i | perl -ne 'if (/DEBUG/ || /#else/) {$skip=1} if (!$skip && !/ifdef DOT_PROD/) {s/^ *//; s/, /,/g; print $_} elsif (/endif/) {$skip=0}' > tmp_data.c + mv tmp_data.c dnn/$i +done + +for i in plc_data.c +do + cat dnn/$i | perl -ne 'if (/#else.*DOT_PROD/) {$skip=1} if (!$skip && !/ifdef DOT_PROD/) {s/^ *//; s/, /,/g; print $_} elsif (/endif.*DOT_PROD/) {$skip=0}' > tmp_data.c + mv tmp_data.c dnn/$i +done |