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.

Sign up for GitHub

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

Jump to bottom

PixelwiseRegressionTask pretrained weights #1306

Merged
merged 1 commit into from
May 4, 2023
Merged

PixelwiseRegressionTask pretrained weights #1306

merged 1 commit into from
May 4, 2023

Conversation

isaaccorley
Copy link
Collaborator

Similarly to #1046, this adds the ability to load pretrained encoder weights from our weight enums and checkpoint files to PixelwiseRegressionTask.

@isaaccorley isaaccorley self-assigned this May 3, 2023
@github-actions github-actions bot added testing Continuous integration testing trainers PyTorch Lightning trainers labels May 3, 2023
@isaaccorley isaaccorley force-pushed the trainers/pixelwiseregression-pretrained-weights branch from 4188a52 to 242a2e2 Compare May 4, 2023 00:54
adamjstewart
adamjstewart approved these changes May 4, 2023
View reviewed changes
Copy link
Collaborator

@adamjstewart adamjstewart left a comment

Choose a reason for hiding this comment

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

Not sure if or where we should clarify that FCN and ViTs are not supported, but it's prob fine.

@adamjstewart adamjstewart added this to the 0.5.0 milestone May 4, 2023
@isaaccorley isaaccorley merged commit d192207 into microsoft:main May 4, 2023
@isaaccorley isaaccorley deleted the trainers/pixelwiseregression-pretrained-weights branch May 4, 2023 02:08
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@adamjstewart adamjstewart adamjstewart approved these changes

Assignees

@isaaccorley isaaccorley

Labels
testing Continuous integration testing trainers PyTorch Lightning trainers
Projects
None yet
Milestone
0.5.0
Development

Successfully merging this pull request may close these issues.
2 participants
@isaaccorley @adamjstewart