mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-09-03 10:06:44 +00:00
[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
This commit is contained in:
@@ -3,23 +3,23 @@
|
||||
|
||||
train_data = dict(
|
||||
dataset=dict(
|
||||
type='CIFAR10Dataset',
|
||||
root='/path/to/data',
|
||||
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))
|
||||
]
|
||||
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',
|
||||
type="DataParallelSampler",
|
||||
shuffle=True,
|
||||
)
|
||||
)
|
||||
),
|
||||
),
|
||||
)
|
||||
|
Reference in New Issue
Block a user