-
Notifications
You must be signed in to change notification settings - Fork 210
Open
Description
Proposal
Add optional SwanLab logging support to SAELens.
Motivation
Some users cannot reliably use Weights & Biases (wandb) due to network restrictions. SwanLab can serve as a drop-in alternative for experiment logging in these environments.
Pitch
- Add
wandb_compat.py, which selectsswanlaborwandbbased onSAE_LENS_LOGGING_BACKEND, and provides a fallbackgenerate_id()implementation (use UUID whenwandb.util.generate_idis unavailable). - Keep calling
wandb.init()/wandb.finish()inllm_sae_training_runner.py, but route the actual backend module through the compatibility layer (either SwanLab or wandb). - When the backend is SwanLab, skip Artifact-related
log_artifactcalls inconfig.py(to avoid unsupported features), and stop usingwandb.Histograminsae_trainer.py.
Checklist
- I have checked that there is no similar [issue](https://github.com/decoderesearch/SAELens/issues) in the repo (required)
Metadata
Metadata
Assignees
Labels
No labels