diff options
author | Jean-Marc Valin <jmvalin@amazon.com> | 2023-09-25 22:29:27 +0300 |
---|---|---|
committer | Jean-Marc Valin <jmvalin@amazon.com> | 2023-09-26 19:11:25 +0300 |
commit | c4b83ae62d84301cbb344e874313eabc38e340a9 (patch) | |
tree | 3c8fee73e469d73df821d7a6b9a04a9222748464 | |
parent | 574c766c0c34eeb01e2c8aec7abd59d667e85267 (diff) |
RDOVAE model update
-rwxr-xr-x | autogen.sh | 2 | ||||
-rw-r--r-- | dnn/torch/rdovae/train_rdovae.py | 2 | ||||
-rw-r--r-- | silk/dred_config.h | 2 |
3 files changed, 3 insertions, 3 deletions
@@ -9,7 +9,7 @@ set -e srcdir=`dirname $0` test -n "$srcdir" && cd "$srcdir" -dnn/download_model.sh fac227e +dnn/download_model.sh 574c766 echo "Updating build configuration files, please wait...." diff --git a/dnn/torch/rdovae/train_rdovae.py b/dnn/torch/rdovae/train_rdovae.py index 3f8484e1..35c0861c 100644 --- a/dnn/torch/rdovae/train_rdovae.py +++ b/dnn/torch/rdovae/train_rdovae.py @@ -231,7 +231,7 @@ if __name__ == '__main__': total_loss = rate_loss + (distortion_loss_hard_quant + distortion_loss_soft_quant) / 2 if args.enable_first_frame_loss: - total_loss = total_loss + 0.5 * torch.relu(first_frame_loss - distortion_loss_hard_quant) + total_loss = .97*total_loss + 0.03 * first_frame_loss total_loss.backward() diff --git a/silk/dred_config.h b/silk/dred_config.h index 891eb1cf..aa06c7ed 100644 --- a/silk/dred_config.h +++ b/silk/dred_config.h @@ -32,7 +32,7 @@ #define DRED_EXTENSION_ID 126 /* Remove these two completely once DRED gets an extension number assigned. */ -#define DRED_EXPERIMENTAL_VERSION 2 +#define DRED_EXPERIMENTAL_VERSION 3 #define DRED_EXPERIMENTAL_BYTES 2 |