How to change the discriminator

Hello author! class NetD is the discriminator, right? Can I use it to replace the model of my own code? My job is image inpainting. If I change the discriminator, what specific operations are required? Should I keep the input and output of my discriminator and just replace the convolutional layer passed in the middle with your class NetD discriminator?