This script is intended to export run metadata from Weights and Biases to Neptune.
This is in beta and we invite feedback and contributions to improve this script.
- A Weights and Biases account,
wandb
library installed, and environment variables set. - A neptune.ai account,
neptune
python library installed, and environment variables set. Read the docs to learn how to set up your installation.
- Run
wandb_to_neptune.py
- Enter the W&B entity name from where you want to export runs, and the Neptune workspace name where you want to import experiments
- Enter the projects you want to export as comma-separated values, without spaces. Enter "all" to export all projects
- Run logs will be created in the same folder as
wandb_to_neptune.py
. You can change this inlogging.basicConfig()
Metadata | W&B | Neptune |
---|---|---|
Project name | example_project | example-project1 |
Project URL | project.url | project.wandb_url |
Run name | run.name | run.sys.name |
Run ID | run.id | run.sys.custom_run_id2 |
Notes | run.notes | run.sys.description |
Tags | run.tags | run.sys.tags |
Config | run.config | run.config3 |
Job type | run.job_type | run.wandb.job_type |
Run path | run.path | run.wandb.path |
Run URL | run.url | run.wandb.url |
Creation time | run.created_at | run.wandb.created_at |
Run summary | run.summary | run.summary3 |
Run metrics | run.scan_history() | run.<METRIC_NAME>4 |
System metrics | run.history(stream="system") | run.monitoring.<METRIC_NAME>5 |
System logs | output.log | run.monitoring.stdout |
Notebook | code/_session_history.ipynb | run.source_code.files |
requirements.txt | requirements.txt | run.source_code.files |
Model checkpoints | *.ckpt/*checkpoint* | run.checkpoints |
Other files | run.files() | run.files |
1 Underscores _
in a W&B project name are replaced by a hyphen -
in Neptune
2 Passing the wandb.run.id as neptune.run.custom_run_id ensures that duplicate Neptune runs are not created for the same W&B run even if the script is run multiple times
3 Values are converted to a string in Neptune
4 _steps
and _timestamps
associated with a metric are logged as step
and timestamp
respectively with a Neptune metric
5 system.
prefix is removed when logging to Neptune
- Models
- W&B table-file†
run.summary
keys starting with_
†- Metrics starting with
_
† - Files with path starting with any of
artifact/
,config.yaml
,media/
,wandb-
†
† These have been excluded at the code level to prevent redundancy and noise, but can be included.
Submit bugs and feature requests as GitHub Issues.
Copyright (c) 2022, Neptune Labs Sp. z o.o.
Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, softwaredistributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License.