ValueError: zero-size array to reduction operation maximum which has no identity

[lbininhbl]

Hi @thangtran480
Every time I ran python train.py, I jus get the error "ValueError: zero-size array to reduction operation maximum which has no identity".

Train all CelebAMask-HQ photos takes several hours, so I reduced the number of images to identify something wrong, but nothing.

I made data struture like this:

Are there any more steps I need to take?

[ductm104]

Did you debug your number of images, mask after load data from dir.

This error occurs when your image is empty.

[ductm104]

Maybe your number of samples is small than BATCH_SIZE, which default is 32. You can try BATCH_SIZE = 2