ok but AFHQ dataset, Four Shapes, MNIST and CIFAR10 are baby datasets; do this on COCO or pascal VOC or imagenet...