diff --git a/autogen.sh b/autogen.sh index b31f74e1e875321b5a3d1db67b7d0912779ba043..b50480e4f24b2b7059f4d9fb46a524cc41ccee61 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 3f8484e1fc578074eba90c607ed3490c27177cde..35c0861cf441c64f734d203d06708b7d5fe7dbe7 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 891eb1cfe0f54dee38138a574787b88b7e9f9fb5..aa06c7ed0f0b8ab87c6879d0e7a710c60c685d97 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