auracaster/piper
3
0
Fork 0
mirror of https://github.com/pstrueb/piper.git synced 2026-04-18 06:15:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

Include discriminator loss in validation

Browse Source
This commit is contained in:
Michael Hansen
2023-02-25 20:08:50 -06:00
parent b8e3058d7a
commit 93d3744614
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/python/larynx_train/vits/lightning.py
View File

@@ -280,7 +280,7 @@ class VitsModel(pl.LightningModule):
return loss_disc_all
def validation_step(self, batch: Batch, batch_idx: int):
val_loss = self.training_step_g(batch)
val_loss = self.training_step_g(batch) + self.training_step_d(batch)
self.log("val_loss", val_loss)
# Generate audio examples