INNER CODE UNIT · TypeScript

checkpoint_path

theaniketgiri/create-llm · src/python-callback-templates.ts:146

        checkpoint_path = self.checkpoint_dir / checkpoint_name
        
        # Create checkpoint
        checkpoint = {
            'step': step,
            'epoch': getattr(trainer, 'epoch', 0),
            'model_state_dict': trainer.model.state_dict(),
            'optimizer_state_dict': trainer.optimizer.state_dict(),
            'loss': loss,
            'config': getattr(trainer, 'config', None),
        }
        
        # Add scheduler if exists
        if hasattr(trainer, 'scheduler') and trainer.scheduler is not None:
            checkpoint['scheduler_state_dict'] = trainer.scheduler.state_dict()
        
        # Save checkpoint
        torch.save(checkpoint, checkpoint_path)

View source record →