Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Description
实现在训练时同步wandb的实验到swanlab,只需在脚本的最开头增加,即可实现跟踪:
测试代码:
Closes: #784
Details
This pull request introduces a new feature to integrate
wandb
(Weights & Biases) withswanlab
by adding a synchronization mechanism. The changes include importing necessary modules, defining thesync_wandb
function, and updating the package's initialization and export lists.Key changes include:
Integration with
wandb
:swanlab/sync/wandb.py
: Added thesync_wandb
function to synchronizewandb
withswanlab
, including patchingwandb.init
,wandb.log
, andwandb.finish
to integrate withswanlab
's logging and configuration management.Module and package updates:
swanlab/__init__.py
: Importedsync_wandb
and updated the export list to includesync_wandb
. [1] [2]swanlab/sync/__init__.py
: Added import and export forsync_wandb
to ensure it is accessible from theswanlab.sync
module.