Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Allow Loading of Checkpoints in Experiments Pipeline #272

Conversation

melo-gonzo
Copy link
Collaborator

This PR addressed #263 for loading checkpoints using the experiments cli. Local or WandB stored checkpoints may be loaded and used for continual training, or encoder weight transfer to a new task. Checkpoints associated with training from the MultiTaskLitModule are processed using matsciml.models.multitask_from_checkpoint which has limited functionality at the moment, and is only suited for continuing training.

@melo-gonzo melo-gonzo requested a review from laserkelvin August 14, 2024 22:16
Copy link
Collaborator

@laserkelvin laserkelvin left a comment

Choose a reason for hiding this comment

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

Looks fantastic!

@laserkelvin laserkelvin added the ux User experience, quality of life changes label Aug 14, 2024
@laserkelvin laserkelvin merged commit 28557a5 into IntelLabs:main Aug 14, 2024
1 check passed
@melo-gonzo melo-gonzo deleted the 263-feature-request-allow-weights-transfer-andor-restarting-from-earlier-checkpoint-with-experiment-cli branch August 22, 2024 21:50
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
ux User experience, quality of life changes
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants