search

skyflynil / stylegan2

StyleGAN2 - Official TensorFlow Implementation with practical improvements
http://arxiv.org/abs/1912.04958
Other
120 stars 33 forks source link

Different batch size for 640x640 dataset #17

Closed Br1lopez closed 3 years ago

Br1lopez commented 4 years ago

Hello, I have been trying to train with my custom dataset (48k, 640x640) and the follwing error appears:

"Cannot batch tensors with different shapes in component 0. First element had shape [3,640,640] and element 1 had shape [1,640,640]"

I have used "--min-h=5 --min-w=5 --res-log2=7".

That first mismatching number is the batch size, isn't it? How could I solve this?

Thanks in advance!

dvschultz commented 4 years ago

first number is channels. this indicates you have a grayscale image amongst some RGB images.