cifar10_noaug 源代码

# Copyright (c) OpenMMLab. All rights reserved.
from mmengine.dataset.sampler import DefaultSampler, InfiniteSampler

from mmagic.datasets.cifar10_dataset import CIFAR10
from mmagic.datasets.transforms.formatting import PackInputs

[文档]cifar_pipeline = [dict(type=PackInputs)]
[文档]cifar_dataset = dict(
    type=CIFAR10,
    data_root='./data',
    data_prefix='cifar10',
    test_mode=False,
    pipeline=cifar_pipeline)

[文档]train_dataloader = dict(
    num_workers=2,
    dataset=cifar_dataset,
    sampler=dict(type=InfiniteSampler, shuffle=True),
    persistent_workers=True)

[文档]val_dataloader = dict(
    batch_size=32,
    num_workers=2,
    dataset=cifar_dataset,
    sampler=dict(type=DefaultSampler, shuffle=False),
    persistent_workers=True)

[文档]test_dataloader = dict(
    batch_size=32,
    num_workers=2,
    dataset=cifar_dataset,
    sampler=dict(type=DefaultSampler, shuffle=False),
    persistent_workers=True)