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

[AutoMM] Support loading specific checkpoints #3244

Merged
merged 1 commit into from
May 30, 2023

Conversation

zhiqiangdon
Copy link
Contributor

Issue #, if available:
#3186

Description of changes:

  • Support loading intermediate checkpoints like predictor = MultiModalPredictor.load("/path/to/a/saved/folder/epoch=3-step=5.ckpt").
  • Refactor predictor and matcher load() to share the same utils.

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

@zhiqiangdon zhiqiangdon added the model list checked You have updated the model list after modifying multimodal unit tests/docs label May 26, 2023
@github-actions
Copy link

Job PR-3244-16e2a9e is done.
Docs are uploaded to http://autogluon-staging.s3-website-us-west-2.amazonaws.com/PR-3244/16e2a9e/index.html

Copy link
Contributor

@FANGAreNotGnu FANGAreNotGnu left a comment

Choose a reason for hiding this comment

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

LGTM

@zhiqiangdon zhiqiangdon merged commit d9e7bec into autogluon:master May 30, 2023
29 of 30 checks passed
@zhiqiangdon zhiqiangdon deleted the mm-load branch May 30, 2023 17:38
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
model list checked You have updated the model list after modifying multimodal unit tests/docs
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants