Skip to content

[Feat] add swanlab logger #1642

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 2 commits into
base: main
Choose a base branch
from
Open

[Feat] add swanlab logger #1642

wants to merge 2 commits into from

Conversation

Zeyi-Lin
Copy link

Pull Request Description

This PR introduces SwanLab, a lightweight open-source experiment tracking tool, as a new logging option for the training framework. The integration provides both online and offline tracking capabilities, along with a local dashboard for visualizing results.

image

We have officially integrated with excellent open-source projects such as transformers, LLaMA Factory, ms-swift, and veRL. We are also very eager to integrate with the outstanding MMEngine to provide developers with a better training experience.

Below is a detailed overview of the changes and usage instructions:

Key Features of SwanLab Integration

1. Online and Offline Tracking:

  • Online Mode: Track experiments remotely and store data on SwanLab's cloud platform.
  • Offline Mode: Use a local dashboard to visualize training logs without an internet connection.

2. Hardware Monitoring:

  • Automatically tracks GPU usage, power consumption, temperature, and other hardware metrics.
  • Supports NVIDIA GPUs and Huawei Ascend NPUs.

3. Remote Access:

  • View training progress remotely via the SwanLab web interface or mobile app.

4. Local Dashboard:

  • Includes an open-source local dashboard for offline visualization of training logs.

This PR is ready for review. Feedback and suggestions are welcome!

@CLAassistant
Copy link

CLAassistant commented Mar 13, 2025

CLA assistant check
All committers have signed the CLA.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants