ColossalAI/tests/test_config/sample_config.py
Hongxin Liu 079bf3cb26
[misc] update pre-commit and run all files (#4752)
* [misc] update pre-commit

* [misc] run pre-commit

* [misc] remove useless configuration files

* [misc] ignore cuda for clang-format
2023-09-19 14:20:26 +08:00

26 lines
632 B
Python

#!/usr/bin/env python
# -*- encoding: utf-8 -*-
train_data = dict(
dataset=dict(
type="CIFAR10Dataset",
root="/path/to/data",
download=True,
transform_pipeline=[
dict(type="RandomResizedCrop", size=224),
dict(type="RandomHorizontalFlip"),
dict(type="ToTensor"),
dict(type="Normalize", mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5)),
],
),
dataloader=dict(
batch_size=64,
pin_memory=True,
num_workers=4,
sampler=dict(
type="DataParallelSampler",
shuffle=True,
),
),
)