def merge_bnorm_test(self):
     merge_bn(self.netG)
     tidy_sequential(self.netG)
 def merge_bnorm_train(self):
     merge_bn(self.netG)
     tidy_sequential(self.netG)
     self.define_optimizer()
     self.define_scheduler()