eden-emu/opus
15
0
Fork 0
mirror of https://github.com/xiph/opus.git synced 2025-06-05 15:03:39 +00:00
Code Issues 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
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
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()