diff --git a/autogen.sh b/autogen.sh index b31f74e1..b50480e4 100755 --- a/autogen.sh +++ b/autogen.sh @@ -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