diff options
author | Jean-Marc Valin <jmvalin@amazon.com> | 2023-10-12 02:19:19 +0300 |
---|---|---|
committer | Jean-Marc Valin <jmvalin@amazon.com> | 2023-10-12 02:19:19 +0300 |
commit | a09e6ca3a1105363a5c6c429a69e9fb46c21306f (patch) | |
tree | 55e43ac6b481c07de32ab46df81f3e2ffd14691e | |
parent | 90c373be08720415a6ae9e01e6dcd46faf82a8e5 (diff) |
quantization
-rw-r--r-- | dnn/torch/fargan/dump_fargan_weights.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/dnn/torch/fargan/dump_fargan_weights.py b/dnn/torch/fargan/dump_fargan_weights.py index 3dfcdb9c..ec1eb9ae 100644 --- a/dnn/torch/fargan/dump_fargan_weights.py +++ b/dnn/torch/fargan/dump_fargan_weights.py @@ -12,7 +12,9 @@ import wexchange.torch import fargan #from models import model_dict -unquantized = [ +unquantized = [ 'cond_net.pembed', 'cond_net.fdense1', 'sig_net.cond_gain_dense', 'sig_net.gain_dense_out' ] + +unquantized2 = [ 'cond_net.pembed', 'cond_net.fdense1', 'cond_net.fconv1', |