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

Full pre-trained model #89

Open
pTomSmash opened this issue Mar 5, 2023 · 1 comment
Open

Full pre-trained model #89

pTomSmash opened this issue Mar 5, 2023 · 1 comment

Comments

@pTomSmash
Copy link

Hi there!
After going over the documentation and playing a bit with the code I understood that:

  1. The PSP model is not pre-trained (Only the conv net that the model loads as a part of the architecture (such as ResNet50).
  2. I therefore cannot just use 'Test' for inference (the code tries to load a training checkpoint that doesn't exist).
  3. I have to train the model myself.

The thing is, I don't have access to such a strong hardware as advised here.
Is there a chance to get the full pre-trained model checkpoint? Am I missing something?

Thanks a lot!

@NovTi
Copy link

NovTi commented Mar 22, 2023

I think the pretrained full model is provided in the repo but it is not that obvious. You can check this link

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

2 participants