mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-04-28 11:45:23 +00:00
079bf3cb26
* [misc] update pre-commit * [misc] run pre-commit * [misc] remove useless configuration files * [misc] ignore cuda for clang-format
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()
|