INNER CODE UNIT · TypeScript

on_validation_end

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

    def on_validation_end(self, trainer: Any, step: int, val_loss: float, val_metrics: Optional[Dict[str, float]] = None):
        """Log validation metrics to TensorBoard"""
        if self.writer is not None:
            self.writer.add_scalar('val/loss', val_loss, step)
            if val_metrics:
                for key, value in val_metrics.items():
                    self.writer.add_scalar(f'val/{key}', value, step)
    
    def on_train_end(self, trainer: Any):
        """Log final metrics and close TensorBoard writer"""
        elapsed = time.time() - self.start_time
        avg_loss = sum(self.losses) / len(self.losses) if self.losses else 0
        
        # Close TensorBoard writer
        if self.writer is not None:
            self.writer.close()
        
        if self.verbose:

View source record →