ref: c4b83ae62d84301cbb344e874313eabc38e340a9
parent: 574c766c0c34eeb01e2c8aec7abd59d667e85267
author: Jean-Marc Valin <jmvalin@amazon.com>
date: Mon Sep 25 11:29:27 EDT 2023
RDOVAE model update
--- a/autogen.sh
+++ b/autogen.sh
@@ -9,7 +9,7 @@
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...."
--- a/dnn/torch/rdovae/train_rdovae.py
+++ b/dnn/torch/rdovae/train_rdovae.py
@@ -231,7 +231,7 @@
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()
--- 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
--
⑨