Error when running Celebahq (edge-to-face)

I follow the setting of 3) Celebahq (edge-to-face), only changing the image size to 128. The error is shown as below:

@panzhang0212 do you have any idea about this error? thanks.

Hi, I run the code with resolution 128, but do not find any problem in the code. What's your command? if you run with 128, you need to add "--load_size 128 ---load_size 128" to the command. load_size can be larger than 128 (for exemple, 156) to do augmentation.

This is the log

Hi, I run the code with resolution 128, but do not find any problem in the code. What's your command? if you run with 128, you need to add "--load_size 128 ---load_size 128" to the command. load_size can be larger than 128 (for exemple, 156) to do augmentation.

Thanks for your reply. For me, the problem is solved by setting the self.preprocess_input(data, ) to self.preprocess_input(data.copy(), ) in line 52 of pix2pix_model.py.

CoCosNet/models/pix2pix_model.py

Line 52 in 3db703a

    
           input_label, input_semantics, real_image, self_ref, ref_image, ref_label, ref_semantics = self.preprocess_input(data, )

This method is effective in the training phase, but needs to be changed back to the data() in the testing phase.
It‘s strange. Can someone explain why?