sagiebenaim / DistanceGAN

Pytorch implementation of "One-Sided Unsupervised Domain Mapping" NIPS 2017

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

distance function in mnist_to_svhn solver

roimehrez opened this issue · comments

Hi, I think you have a small bug in the mnist_to_svhn/solver (line 94)

def distance(self, A, B): return torch.abs(torch.mean(A) - torch.mean(B))

should be

def distance(self, A, B): return torch.mean(torch.abs(A - B))

thanks!