(self, epoch)
Save parameters to checkpoint
Save parameters to checkpoint
Save parameters to checkpoint
def save_model(self, epoch):
        """Save parameters to checkpoint"""
        ckpt_path = os.path.join(self.config.save_path, f'{epoch}.pkl')
        print(f'Save parameters to {ckpt_path}')
        torch.save(self.model.state_dict(), ckpt_path)
def save_model ( self , epoch ) : ckpt_path = os . path . join ( self . config . save_path , f'{epoch}.pkl' ) print ( f'Save parameters to {ckpt_path}' ) torch . save ( self . model . state_dict ( ) , ckpt_path )