mirror of https://github.com/hpcaitech/ColossalAI
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
14 lines
265 B
14 lines
265 B
2 years ago
|
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()
|