Merge pull request #782 from AznamirWoW/audio_tensorboard

IAHispano · Oct 4, 2024 · 2062949 · 2062949
2 parents 393cf0d + ce5864c
commit 2062949
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/rvc/train/train.py b/rvc/train/train.py
@@ -779,7 +779,8 @@ 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)
+                    with torch.no_grad():
+                        o, *_ = net_g.infer(phone, phone_lengths, pitch, pitchf, sid)
                     audio_dict.update({f"gen/audio_{global_step:07d}": o[0, :, : ]})
 
                     summarize(