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

fix config

pull/58/head
hezijian 2020-10-21 14:16:18 +08:00
parent d542115530
commit 8e996175b1
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
configs/selfsup/byol/r50_bs256_accumulate16_ep300.py
View File

@ -33,8 +33,8 @@ data_source_cfg = dict(
type='ImageNet',
memcached=True,
mclient_path='/mnt/lustre/share/memcached_client')
data_train_list = '/data/ImageNet/meta/train.txt'
data_train_root = '/data/ImageNet/train'
data_train_list = 'data/imagenet/meta/train.txt'
data_train_root = 'data/imagenet/train'
dataset_type = 'BYOLDataset'
img_norm_cfg = dict(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
train_pipeline = [
@ -72,7 +72,7 @@ train_pipeline2[4]['p'] = 0.1 # gaussian blur
train_pipeline2[5]['p'] = 0.2 # solarization
data = dict(
imgs_per_gpu=16, # total 32*8(gpu)*16(interval)=4096
imgs_per_gpu=32, # total 32*8(gpu)*16(interval)=4096
workers_per_gpu=4,
train=dict(
type=dataset_type,