INNER CODE UNIT · TypeScript
elapsed
theaniketgiri/create-llm · src/python-callback-templates.ts:314
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:
print(f"{'='*60}")
print(f"Training completed!")
print(f"Total time: {self._format_time(elapsed)}")
print(f"Average loss: {avg_loss:.4f}")
print(f"{'='*60}")
def _log_metrics(self, trainer: Any, step: int, loss: float, metrics: Optional[dict]):
"""Log metrics to console, file, and TensorBoard"""
elapsed = time.time() - self.start_time