How to train the User-Guided model?

[JinMengKang]

Dear author, I am very interested in the user - guide image in painting part of your work. However, I found that if I train directly, there is no image of EDG. How should I train such a model? What part of the code should I change.
Another question, is your HED a pre-training model? Can you tell me where to modify the code that has been converted to TensorFlow in your code
Looking forward to your reply.

[avalonstrel]

Thank you for your question, I have transfer the code into pytorch version, and you can refer to https://github.com/avalonstrel/GatedConvolution_pytorch for a better implementation. I am sorry to say I just abandon this version.