When training with the celeba dataset, the program encountered an error. 