⬅ datasets/pyg_colored_mnist_dataset.py