Skip to content
This repository has been archived by the owner on Dec 16, 2022. It is now read-only.

make W&B callback resumable #5312

Merged
merged 3 commits into from
Jul 19, 2021
Merged

make W&B callback resumable #5312

merged 3 commits into from
Jul 19, 2021

Conversation

epwalsh
Copy link
Member

@epwalsh epwalsh commented Jul 14, 2021

Ensures the Weights & Biases callback will resume logging to the same run on allennlp train --resume .... See https://docs.wandb.ai/guides/track/advanced/resuming for reference. Here's an example run that was stopped and then restarted: https://wandb.ai/allenai-team1/allennlp-testing/runs/2gf9ug4m/overview.

@epwalsh epwalsh requested a review from dirkgr July 14, 2021 21:47
Comment on lines +164 to +167
def state_dict(self) -> Dict[str, Any]:
return {
"run_id": self._run_id,
}
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I love it when a plan comes together. I'm not sure we had any callbacks using state dicts when we put in support for those?

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Yeup!

@dirkgr dirkgr enabled auto-merge (squash) July 19, 2021 21:46
@dirkgr dirkgr merged commit ef5400d into main Jul 19, 2021
@dirkgr dirkgr deleted the wandb-resume branch July 19, 2021 22:37
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants