def get_current_dis(self):
"""Return the distribution of encoder features"""
dis_ret = OrderedDict()
value = getattr(self, 'distribution')
for i in range(1):
for j, name in enumerate(self.value_names):
if isinstance(name, str):
dis_ret[name+str(i)] =util.tensor2array(value[i][j].data)
return dis_ret
def get_current_visuals(self):
real_A = util.tensor2array(self.real_A.data)
fake_B = util.tensor2array(self.fake_B.data)
real_B = util.tensor2array(self.real_B.data)
return OrderedDict([('real_A', real_A), ('fake_B', fake_B), ('real_B', real_B)])
