Could you provide the trained model?

[JANEKKO]

Thanks for your code ,but our gpu is only 4GB in my lab , Could you provide the trained model? I need to solve some problem in industry filed. Sorry for borther you.

[JANEKKO]

I learned that this code need at least 11GB Gpu

[KyrieLiu52]

The author provides the download link of pretrained model in the readme file, you can test with pretrained model first.

[JANEKKO]

Thank you. I will try it firstly.