Unverified Commit 6178b7d6 authored by Phil Wang's avatar Phil Wang Committed by GitHub
Browse files

Merge pull request #37 from djqualia/patch-3 

Make the training sample output filename unique
parents d4e278b6 7f3b2d68

audiolm_pytorch/trainer.py

+1 −1
Original line number Diff line number Diff line
@@ -345,7 +345,7 @@ class SoundStreamTrainer(nn.Module): 
                milestone = steps // self.save_results_every



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

                    filename = str(self.results_folder / f'sample.flac')

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

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



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