INNER CODE UNIT · TypeScript

log_path

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

        log_path = self.log_dir / self.log_file
        with open(log_path, 'w') as f:
            f.write("step,loss,lr,tokens_per_sec,elapsed_time\\n")
        
        if self.verbose:
            print(f"Training logs will be saved to: {log_path}")
            print(f"{'='*60}")
            print(f"{'Step':<10} {'Loss':<12} {'LR':<12} {'Tokens/s':<12} {'Time'}")
            print(f"{'='*60}")
    
    def on_step_end(self, trainer: Any, step: int, loss: float, metrics: Optional[Dict[str, float]] = None):
        """Log metrics at intervals"""
        self.losses.append(loss)
        
        if step % self.log_interval == 0 and step > 0:
            self._log_metrics(trainer, step, loss, metrics)
    
    def on_validation_end(self, trainer: Any, step: int, val_loss: float, val_metrics: Optional[Dict[str, float]] = None):

View source record →