_base_ = '../imagenet/rotation-pred_sz224_bs64.py'

# dataset settings
data_source_cfg = dict(type='ImageNet', return_label=False)
# ImageNet dataset, 100 class
data_train_list = 'data/meta/ImageNet100/train.txt'
data_train_root = 'data/ImageNet/train'
data_test_list = 'data/meta/ImageNet100/val.txt'
data_test_root = 'data/ImageNet/val/'

# dataset summary
data = dict(
    train=dict(
        data_source=dict(
            list_file=data_train_list, root=data_train_root,
            **data_source_cfg),
    ),
    val=dict(
        data_source=dict(
            list_file=data_test_list, root=data_test_root,
            **data_source_cfg),
    ),
)
