feature: update wandb callback to add checkpoints#1

Closed

parambharat wants to merge 8 commits intomainfrom

feature/update-wandb-callback

Commits on Jan 4, 2023

docs: update docstrings to fix broken links and changes
Bharat Ramanathan
committed
fix: change default WANDB_WATCH behavior to false.
Bharat Ramanathan
committed
feature: add on_save method to save model checkpoints
Bharat Ramanathan
committed
format: remove unused formatted string from logging
Bharat Ramanathan
committed

Commits on Jan 5, 2023

format: fixup, linting and black formatting fixes
Bharat Ramanathan
committed
merge: update branch with upstream changes
Bharat Ramanathan
committed
feature: change run name to wandb autogenerated name
Bharat Ramanathan
committed

Commits on Jan 6, 2023

fix: change checkpoint to artifact type `model`
Bharat Ramanathan
committed