Files
ColossalAI/tests/test_data/test_cifar10_dataset.py