tensorboard

Visualize training metrics, model graphs, and performance profiles with TensorBoard.

13.0k|930|Updated Nov 3, 2025
One-click install
npx skills add https://github.com/Orchestra-Research/AI-research-SKILLs --skill tensorboard-orchestra-research
Or copy as Structured Prompt for Agent▼
Please help me install this Agent Skill.
Skill: tensorboard
Source: https://github.com/Orchestra-Research/AI-research-SKILLs/tree/main/13-mlops/tensorboard
Command: npx skills add https://github.com/Orchestra-Research/AI-research-SKILLs --skill tensorboard-orchestra-research

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill requires tensorboard, torch, tensorflow, and includes references (resource) components.

What problem does it solve? Training machine learning models without visibility into metrics, weight distributions, and hardware utilization makes debugging slow and experiment comparison difficult. This Skill provides complete guidance for instrumenting PyTorch and TensorFlow training loops with TensorBoard logging. ## Core Features & Use Cases - Metric & Experiment Tracking: Log scalars, images, histograms, text, and PR curves, then compare multiple runs side-by-side in one dashboard. - Model Debugging: Visualize model graphs, weight and gradient histograms, activation distributions, and high-dimensional embeddings with PCA, t-SNE, or UMAP projection. - Performance Profiling: Profile CPU and GPU operations, track memory allocation, and identify bottlenecks using the PyTorch and TensorFlow profilers. - Use Case: While fine-tuning a ResNet model, log training and validation loss per epoch, weight histograms every 5 epochs, and sample predictions every 10 epochs, then launch TensorBoard to compare runs across learning rates. ## Quick Start Instrument my PyTorch training loop with TensorBoard logging for loss, accuracy, and weight histograms, then show me how to launch the dashboard.

Frequently Asked Questions about tensorboard

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I log training metrics to TensorBoard in PyTorch?▼

Create a SummaryWriter from torch.utils.tensorboard with a log directory, then call writer.add_scalar with a tag, value, and step inside your training loop. Launch the dashboard with tensorboard --logdir=runs and open http://localhost:6006.

How to compare multiple experiments in TensorBoard?▼

Write each run to a separate subdirectory under a common log directory, such as runs/exp1 and runs/exp2. Launch TensorBoard on the parent directory and all runs appear together, where you can toggle, filter with regex, and overlay charts.

Does TensorBoard work with HuggingFace Transformers?▼

Yes, set report_to='tensorboard' and logging_dir in TrainingArguments, and the Trainer automatically logs metrics. You can also attach a custom TrainerCallback with a SummaryWriter for manual logging of additional values.

Can I profile GPU performance with TensorBoard?▼

Yes, use torch.profiler with the tensorboard_trace_handler for PyTorch, or set profile_batch in the Keras TensorBoard callback for TensorFlow. The Profile tab shows GPU utilization, kernel stats, memory usage, and operation timelines.

Why do TensorBoard log files grow too large?▼

Logging every batch creates excessive event files. Log epoch-level metrics every epoch and batch-level metrics only periodically, such as every 100 batches, and log histograms every few epochs instead of every step.

How do I visualize embeddings in TensorBoard?▼

Call writer.add_embedding with your embedding tensor, optional metadata labels, and label images. Open the Projector tab to explore the data with PCA, t-SNE, or UMAP, and color or filter points by their metadata labels.