Unverified Commit 75d71ad4 authored by Phil Wang's avatar Phil Wang Committed by GitHub
Browse files

Merge pull request #71 from djqualia/patch-11 

Fix training output sample hz in non-default cases
parents 691ea517 7e20bf98

audiolm_pytorch/trainer.py

+1 −1
Original line number Diff line number Diff line
@@ -392,7 +392,7 @@ class SoundStreamTrainer(nn.Module): 


                for ind, recon in enumerate(recons.unbind(dim = 0)):

                    filename = str(self.results_folder / f'sample_{steps}.flac')

                    torchaudio.save(filename, recon.cpu().detach(), DEFAULT_SAMPLE_RATE)

                    torchaudio.save(filename, recon.cpu().detach(), self.soundstream.target_sample_hz)



            self.print(f'{steps}: saving to {str(self.results_folder)}')