def define_module(self):
ngf = self.gf_dim
print ("Calling ATT_NET with idf = ", ngf,"cdf = ", self.ef_dim)
self.att = ATT_NET(ngf, self.ef_dim)
print("back at NEXT_STAGE_G")
self.residual = self._make_layer(ResBlock, ngf * 2)
print("self.residual", self.residual)
self.upsample = upBlock(ngf * 2, ngf)
print("self.upsample", self.upsample)
def define_module(self):
ngf = self.gf_dim
self.att = ATT_NET(ngf, self.ef_dim)
self.residual = self._make_layer(ResBlock, ngf * 2)
self.upsample = upBlock(ngf * 2, ngf)