Unverified Commit 43def8e0 authored by Taylor's avatar Taylor Committed by GitHub
Browse files

Correct coarse trainer ckpt filename 

Coarse trainer was saving checkpoints mislabeled as fine
parent 4bcce75a

audiolm_pytorch/trainer.py

+1 −1
Original line number Diff line number Diff line
@@ -963,7 +963,7 @@ class CoarseTransformerTrainer(nn.Module): 
        # save model every so often



        if self.is_main and not (steps % self.save_model_every):

            model_path = str(self.results_folder / f'fine.transformer.{steps}.pt')

            model_path = str(self.results_folder / f'coarse.transformer.{steps}.pt')

            self.save(model_path)



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