[Fix]: Fix mask generator name bug

open-mmlab · Apr 22, 2022 · 7a00f4f · 7a00f4f
1 parent 9cd0ae2
commit 7a00f4f
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/configs/selfsup/_base_/datasets/imagenet_cae.py b/configs/selfsup/_base_/datasets/imagenet_cae.py
@@ -20,13 +20,18 @@
 
 train_pipeline.append(
     dict(
-        type='MaskingGenerator',
+        type='BEiTMaskGenerator',
         input_size=(14, 14),
         num_masking_patches=75,
         max_num_patches=None,
         min_num_patches=16))
 
 # dataset summary
+file_client_args = dict(
+    backend='memcached',
+    server_list_cfg='/mnt/lustre/share/memcached_client/pcs_server_list.conf',
+    client_cfg='/mnt/lustre/share_data/zhangwenwei/software/pymc/mc.conf',
+    sys_path='/mnt/lustre/share_data/zhangwenwei/software/pymc')
 data = dict(
     samples_per_gpu=256,
     workers_per_gpu=8,
@@ -35,6 +40,7 @@
         data_source=dict(
             type=data_source,
             data_prefix='data/imagenet/train',
-            ann_file='data/imagenet/meta/train.txt'),
+            ann_file='data/imagenet/meta/train.txt',
+            file_client_args=file_client_args),
         pipeline=train_pipeline,
         prefetch=prefetch))