mirror of https://github.com/hpcaitech/ColossalAI
14 lines
265 B
Python
14 lines
265 B
Python
|
import os
|
||
|
|
|
||
|
|
from torchvision.datasets import CIFAR10
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
dir_path = os.path.dirname(os.path.realpath(__file__))
|
||
|
|
data_root = os.path.join(dir_path, 'data')
|
||
|
|
dataset = CIFAR10(root=data_root, download=True)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main()
|