Code seems incomplete..?

[JawnWB]

Hi, i've tried to train my own model, but the code seems incomplete and/or I'm not doing something correctly.

which_model_G: 'inpainting_resnet'

(which is the value in the example cfgs from github) gives the error --

NotImplementedError: Unsupported generator model: inpainting_resnet

It seems networks.py expects inpainting_resnet

If change that, then I get the error --

self.netD = networks.define_D(opt).to(self.device)

AttributeError: module 'models.networks' has no attribute 'define_D'

I can't see a discriminator model in the models directory, so I wonder if you have any advice?

kind regards John

[Zheng222]

@JawnWB I have fixed this issue in models.networks.define_D(). Thanks for your question.