xtils.get_data_root(data='cifar10'),
'data_augment': {
'train': '1crop-flip',
'val': 'no-aug'
},
'data_kwargs': {},
'data_workers':
4,
# path config
'current_time':
'',
'ckpt_suffix':
'', # when save a ckpt, u can add a special mark to its filename.
'ckpt_base_dir':
xtils.get_base_dir(k='ckpt'),
'ckpt_dir':
'auto-setting',
'log_base_dir':
xtils.get_base_dir(k='log'),
'log_dir':
'auto-setting',
# iter config
'start_iter':
0,
'max_iters': [100 * BN, 90 * BN, 60 * BN, 40 * BN, 120 * BN][0],
'start_epoch':
0,
'max_epochs':
0,
'mgpus_to_sxpu': ['m2s', 's2m', 'none', 'auto'][3],
# data config
'dataset': 'imagenet',
'data_info': {'train_size': train_size, 'val_size': 10000, 'test_size': 10000},
'data_root': xtils.get_data_root(data='imagenet'),
'data_augment': {'train': 'rotate-rresize-1crop', 'val': '1resize-1crop',
'imsize': 256, 'insize': 224, 'color': True, 'interp': 'bilinear',
'degree': (0, 0), 'scale': (0.08, 1), 'ratio': (3. / 4, 4. / 3)},
'data_kwargs': {},
'data_workers': 24,
# path config
'current_time': '',
'ckpt_suffix': '', # when save a ckpt, u can add a special mark to its filename.
'ckpt_base_dir': xtils.get_base_dir(k='ckpt'),
'ckpt_dir': 'auto-setting',
'log_base_dir': xtils.get_base_dir(k='log'),
'log_dir': 'auto-setting',
# iter config
'start_iter': 0,
'max_iters': [100 * BN, 90 * BN, 140 * BN, 420 * BN][-1],
'start_epoch': 0,
'max_epochs': 0,
'bsize_train': batch_size,
'bsize_val': batch_size_val,
'batch_nums': batch_nums,
'Unit': {'epoch': batch_nums, 'iter': 1}[['epoch', 'iter'][0]], # 按epoch为单位调节 还是按iteration为单位调节lr/bs？
'BN': BN,