grayscale is now inferred from options if it is not explicitly specified

taesungp / contrastive-unpaired-translation

Contrastive unpaired image-to-image translation, faster and lighter training than cyclegan (ECCV 2020, in PyTorch)

https://taesung.me/ContrastiveUnpairedTranslation/

Other

2.23k stars 417 forks source link

grayscale is now inferred from options if it is not explicitly specified #109

Open IljaManakov opened 3 years ago

IljaManakov commented 3 years ago

Fixes #46

andreasoie commented 2 years ago

Have you even tested this? Because I don't think it'll work since:

File "../../CUT/models/cut_model.py", line 151, in forward                                    │
    self.real = torch.cat((self.real_A, self.real_B), dim=0) if self.opt.nce_idt and self.opt.isTrain els│
e self.real_A                                                                                            │
RuntimeError: Sizes of tensors must match except in dimension 0. Expected size 3 but got size 1 for tenso│
r number 1 in the list.