Wrong size of dataloader

[konatasick]

When I run the wgan training, I set the dataset as mnist, and the dataloader returns data like: [64,1,28, 28].

But when I set it as Cifar10, it returns [64, 32, 32, 3].

Is there something wrong with the paddlegan dataloader or I should always swap the dimension when using RGB images?

[qingqing01]

@konatasick Maybe for Cifar10, you need to other data transform.

[JunnYu]

问题过于久远，如果有图像和视频生成的需求，可以使用新的跨模态工具: https://github.com/PaddlePaddle/PaddleMIX/tree/develop