
SwanLab Experiment Tracking
Lets the agent set up SwanLab to log, compare and visualise ML training runs locally or self-hosted.
- Repository
- Orchestra-Research/AI-Research-SKILLs
- Stars
- 14,195
- Licence
- MIT
- Files
- 3 files
What it does
When installed, the agent knows how to instrument machine learning training code with SwanLab, an open-source experiment tracker. It can add code that records metrics, hyperparameter configs, images, audio, text and custom charts for each run, and can set up a local or self-hosted dashboard so results are viewed in a browser rather than only in a managed cloud service. It also covers hooking SwanLab into frameworks such as PyTorch, HuggingFace Transformers, PyTorch Lightning and Fastai.
Great for
- ML engineers who want to track and compare training runs without relying on a managed SaaS
- Researchers who need local or self-hosted logging for private or offline experiments
- Developers adding experiment logging to an existing PyTorch or Transformers training script
Skip it if
- Teams already standardised on another tracker (such as Weights & Biases or MLflow) who only need help with that tool, since this skill covers SwanLab only
- Non-technical users with no training code to instrument, because the skill is about writing and running Python logging code
Try asking
- “Add SwanLab logging to my PyTorch training loop so train and val loss are recorded each epoch”
- “Set up local-only SwanLab tracking in ./swanlog and show me how to open the dashboard with swanlab watch”
- “Configure my HuggingFace Trainer to report to SwanLab with a run name”
- “Log sample images and a loss curve chart to my SwanLab run”
Install in 3 steps
- 1
Download the skill
The ZIP contains the 3 original files, the MIT licence and an attribution note.
swanlab.zip - 2
Add it to your agent
Claude Code (all projects)
mkdir -p ~/.claude/skills && unzip swanlab.zip -d ~/.claude/skills/Claude Code / Agent SDK (this project only)
mkdir -p .claude/skills && unzip swanlab.zip -d .claude/skills/Claude apps (web/desktop): Settings → Capabilities → Skills → Upload skill, and choose the ZIP.
- 3
Ask for the task
No special command needed — the agent reads the skill's description and loads it when your request matches. Try one of the prompts above.
Get better results
- Use mode="local" with a logdir when you do not want data sent to any server.
- Store the SWANLAB_API_KEY in an environment variable rather than pasting it into code or chat.
- Review the generated training code before running it, since the agent will execute scripts and install packages in your environment.
What's inside
- references/integrations.md
- references/visualization.md
- SKILL.md
Redistributed unmodified from Orchestra-Research/AI-Research-SKILLs under the MIT licence. The assessment was drafted with AI from the skill's files on 2026-10-29. Skills can instruct your agent to run code — read SKILL.md and any scripts before installing.