diff --git a/mmpretrain/configs/_base_/datasets/imagenet_bs32.py b/mmpretrain/configs/_base_/datasets/imagenet_bs32.py
index 63e2d975..7d074008 100644
--- a/mmpretrain/configs/_base_/datasets/imagenet_bs32.py
+++ b/mmpretrain/configs/_base_/datasets/imagenet_bs32.py
@@ -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),
diff --git a/tools/train.py b/tools/train.py
index 046e9d66..84c1eec9 100644
--- a/tools/train.py
+++ b/tools/train.py
@@ -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)