Merge pull request #776 from AznamirWoW/audio_tensorboard

added audio output for tensorboard
IAHispano · Oct 3, 2024 · 86da712 · 86da712
2 parents 9a8a305 + 921d11e
commit 86da712
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/rvc/train/train.py b/rvc/train/train.py
@@ -778,11 +778,17 @@ def train_and_evaluate(
                         ),
                         "all/mel": plot_spectrogram_to_numpy(mel[0].data.cpu().numpy()),
                     }
+                    audio_dict = {}
+                    o, *_ = net_g.infer(phone, phone_lengths, pitch, pitchf, sid)
+                    audio_dict.update({f"gen/audio_{global_step:07d}": o[0, :, : ]})
+
                     summarize(
                         writer=writer,
                         global_step=global_step,
                         images=image_dict,
                         scalars=scalar_dict,
+                        audios=audio_dict,
+                        audio_sample_rate=config.data.sample_rate,
                     )
 
             global_step += 1