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)