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

Make the training sample output filename unique 

Without this, "sample.flac" keeps getting overwritten, and it's nice to have a history of how the training is progressing.
parent d4e278b6

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