Swanlab
SwanLab is an open-source AI experiment tracking and visualization tool designed to support researchers and teams in managing deep learning model training. It offers a platform to track, record, and compare experiments with support for over 30 mainstream AI training frameworks, including HuggingFace Transformers and PyTorch Lightning. SwanLab provides both cloud and offline usage modes, enabling flexibility for different development environments. The tool includes a Python API for logging hyperparameters, metrics, and multimedia content such as images and audio, facilitating detailed experiment documentation. The platform features an interactive dashboard with charts for visualizing training metrics, system resources, and experiment comparisons. It supports multi-user collaboration through online sharing within organizations and integrates with development environments via a VSCode plugin. SwanLab's open-source nature allows users to self-host and customize their setup, with local visualization available through an additional dashboard extension.
SwanLab is an open-source AI experiment tracking tool supporting cloud and offline modes with multi-framework integration and collaboration features.
AI Research Experiment Tracking
Researchers conducting deep learning experiments can log hyperparameters, metrics, and multimedia data to monitor training progress and compare results.
Team Collaboration on Model Development
Teams can share experiment results via links and collaborate on model training projects using SwanLab's multi-user features.
Offline Experiment Management
Users working in environments without reliable internet can use SwanLab's local and offline modes to track and visualize experiments.
pip install swanlab to install the core package.swanlab login -k api-key or swanlab.login(api_key="Your API Key") with a key obtained from https://swanlab.cn settings.run = swanlab.init(project="my-project", config={"learning_rate": 0.01}).run.log({"loss": loss}).swanlab watch ./swanlog after installing the dashboard extension.