semseg icon indicating copy to clipboard operation
semseg copied to clipboard

Full pre-trained model

Open pTomSmash opened this issue 1 year ago • 1 comments

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!

pTomSmash avatar Mar 05 '23 13:03 pTomSmash

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

NovTi avatar Mar 22 '23 04:03 NovTi