diff options
author | Jan Buethe <jbuethe@amazon.de> | 2023-12-18 14:33:11 +0300 |
---|---|---|
committer | Jan Buethe <jbuethe@amazon.de> | 2024-01-19 18:42:54 +0300 |
commit | 692196fcb58bc5f0135e603da83f60fb966a70c1 (patch) | |
tree | fa39d4514bebb45b55af0f775bc32bfcb60bbfbf /dnn/torch/osce/utils/layers | |
parent | bfcbbbcb38ce399a6f4cf629c80097bc7d1dadbc (diff) |
bugfix
Diffstat (limited to 'dnn/torch/osce/utils/layers')
-rw-r--r-- | dnn/torch/osce/utils/layers/td_shaper.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/dnn/torch/osce/utils/layers/td_shaper.py b/dnn/torch/osce/utils/layers/td_shaper.py index 7080a1b3..0d2052ee 100644 --- a/dnn/torch/osce/utils/layers/td_shaper.py +++ b/dnn/torch/osce/utils/layers/td_shaper.py @@ -52,7 +52,7 @@ class TDShaper(nn.Module): self.feature_alpha1_t = nn.Conv1d(self.env_dim, frame_size, 2) self.feature_alpha2 = nn.Conv1d(frame_size, frame_size, 2) - if soft_quant: + if softquant: self.feature_alpha1_f = soft_quant(self.feature_alpha1_f) if self.innovate: |