eden-emu/opus
15
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 4

RDOVAE model update

Browse source
This commit is contained in:
Jean-Marc Valin 2023-09-25 15:29:27 -04:00
parent 574c766c0c
commit c4b83ae62d
No known key found for this signature in database
GPG key ID: 531A52533318F00A
3 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
autogen.sh
View file

@ -9,7 +9,7 @@ set -e
srcdir=`dirname $0` srcdir=`dirname $0`
test -n "$srcdir" && cd "$srcdir" test -n "$srcdir" && cd "$srcdir"
dnn/download_model.sh fac227e dnn/download_model.sh 574c766
echo "Updating build configuration files, please wait...." echo "Updating build configuration files, please wait...."

2
dnn/torch/rdovae/train_rdovae.py
View file

@ -231,7 +231,7 @@ if __name__ == '__main__':
total_loss = rate_loss + (distortion_loss_hard_quant + distortion_loss_soft_quant) / 2 total_loss = rate_loss + (distortion_loss_hard_quant + distortion_loss_soft_quant) / 2
if args.enable_first_frame_loss: 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() total_loss.backward()

2
silk/dred_config.h
View file

@ -32,7 +32,7 @@
#define DRED_EXTENSION_ID 126 #define DRED_EXTENSION_ID 126
/* Remove these two completely once DRED gets an extension number assigned. */ /* 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 #define DRED_EXPERIMENTAL_BYTES 2