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

how use custom pre-trained backbone in mask2former #226

Open
sleepat-11 opened this issue Jan 18, 2024 · 1 comment
Open

how use custom pre-trained backbone in mask2former #226

sleepat-11 opened this issue Jan 18, 2024 · 1 comment

Comments

@sleepat-11
Copy link

Your work is excellent and impressive!
But I have a little question how to load my own pre-trained backbone to replace the origin backbone, for example .i got a vit pre-trained model ,which is a pth or pt file, how to load pth in detectron2? do i have to tranfer it into pkl file by convert-torchvision-to-d2.py?
Thanks again for your great work!

@sleepat-11
Copy link
Author

and i also need to freeze my backbone parameters? how to do that?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant