Unverified Commit 7e20bf98 authored by djqualia's avatar djqualia Committed by GitHub
Browse files

Fix training output sample hz in non-default cases 

Previous code assumed output was 16000....
parent 691ea517

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)}')