Skip to content

[WIP] feat: Add generic Wandb integration#180

Merged
MarcosRiveraMartinez merged 1 commit intostochasticai:mainfrom
cnbeining:wandb
May 8, 2023
Merged

[WIP] feat: Add generic Wandb integration#180
MarcosRiveraMartinez merged 1 commit intostochasticai:mainfrom
cnbeining:wandb

Conversation

@cnbeining
Copy link
Copy Markdown
Contributor

WIP but with high confidence.

Tests welcomed for non-LoRA training as I don't have powerful GPU :-(

Usage:

from pytorch_lightning.loggers import WandbLogger  # Or any other logger supported by pytorch_lightning

# Initializes WandB integration 
wandb_logger = WandbLogger()

dataset = Dataset("whatever")
model = BaseModel.create("whatever")

model.finetune(dataset=dataset, logger=wandb_logger)

@MarcosRiveraMartinez MarcosRiveraMartinez merged commit 21f7902 into stochasticai:main May 8, 2023
glennko pushed a commit that referenced this pull request Sep 28, 2025
[WIP] feat: Add generic Wandb integration
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