ewrfcas/MST_inpainting

How to generate SharedWEModel_gen.py?

kopop8 opened this issue · 3 comments

Hello!
I just trained the network on both stages on my own dataset, with good looking result! Now I want to test a single image by running test_single.py. test_single.py however uses SharedWEModel_gen.py that was not generated while training my network. Am I missing something to generate this? Or do I need to change it to another checkpoint?

I can specify more info if needed but right now it seems I just missed something.

Would really appreciate some help 😄! Overall really awesome project can't wait for the new one to come out.

Hi, since the training code is rewritten, there are some differences compared with the test code.
Differences have been shown in the README, and you can check about it.

Okay, I will rewrite the test code, thanks!

好的,我重新寫一下測試代碼,謝謝!

Hello, I have the same problem as you, can you provide your modified test code?
I have modified it for a long time and still have problems, please trouble you, thank you.