yunjey / stargan

StarGAN - Official PyTorch Implementation (CVPR 2018)
MIT License
5.21k stars 967 forks source link

zero vector order in the domain label vector #135

Open yy97831 opened 3 years ago

yy97831 commented 3 years ago

Hi can i change the order of the zero vector in the domain label from : c_org = torch.cat([zero, c_org, mask], dim=1)
c_trg = torch.cat([zero, c_trg, mask], dim=1)

to

c_org = torch.cat([c_org, zero, mask], dim=1)
c_trg = torch.cat([ c_trg, zero, mask], dim=1)

it should be the same as the network will ignore the zeros anyway, am I right?

thanks