mirrors
/
mmpretrain
mirror of https://github.com/open-mmlab/mmpretrain.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

keep other files no change

pull/1727/head
zeyuanyin 2023-07-26 18:46:58 +04:00
parent 743ed01cdf
commit 48d5ca95ef
2 changed files with 2 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mmpretrain/configs/_base_/datasets/imagenet_bs32.py
View File

@ -38,7 +38,7 @@ train_dataloader = dict(
dataset=dict(
type=dataset_type,
data_root='data/imagenet',
# ann_file='meta/train.txt',
ann_file='meta/train.txt',
data_prefix='train',
pipeline=train_pipeline),
sampler=dict(type=DefaultSampler, shuffle=True),
@ -50,7 +50,7 @@ val_dataloader = dict(
dataset=dict(
type=dataset_type,
data_root='data/imagenet',
# ann_file='meta/val.txt',
ann_file='meta/val.txt',
data_prefix='val',
pipeline=test_pipeline),
sampler=dict(type=DefaultSampler, shuffle=False),

12
tools/train.py
View File

@ -146,18 +146,6 @@ def main():
# load config
cfg = Config.fromfile(args.config)
# print('default train data root: ', cfg.train_dataloader.dataset.data_root)
# print('default val data root: ', cfg.val_dataloader.dataset.data_root)
cfg.train_dataloader.dataset.data_root = '/home/zeyuan.yin/imagenet'
cfg.val_dataloader.dataset.data_root = '/home/zeyuan.yin/imagenet'
print('dataset cfg', cfg.train_dataloader.dataset)
print('---')
# print('model cfg', cfg.model)
# print('optim_wrapper cfg', cfg.optim_wrapper)
exit()
# merge cli arguments to config
cfg = merge_args(cfg, args)