mirrors/mmselfsup
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmselfsup.git synced 2025-06-03 14:59:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

update places205 linear classification config

Browse Source
This commit is contained in:
Jiahao000 2020-09-01 00:06:55 +08:00
parent 30c14b415e
commit c3f7d6adf6
2 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
configs/benchmarks/linear_classification/places205/r50_multihead.py
View File

@ -46,7 +46,7 @@ test_pipeline = [
] ]
data = dict( data = dict(
imgs_per_gpu=32, # total 32x8=256 imgs_per_gpu=32, # total 32x8=256
workers_per_gpu=5, workers_per_gpu=4,
train=dict( train=dict(
type=dataset_type, type=dataset_type,
data_source=dict( data_source=dict(
@ -65,7 +65,7 @@ custom_hooks = [
dataset=data['val'], dataset=data['val'],
initial=True, initial=True,
interval=10, interval=10,
imgs_per_gpu=128, imgs_per_gpu=32,
workers_per_gpu=4, workers_per_gpu=4,
eval_param=dict(topk=(1, ))) eval_param=dict(topk=(1, )))
] ]
@ -78,7 +78,7 @@ optimizer = dict(
paramwise_options=dict(norm_decay_mult=0.), paramwise_options=dict(norm_decay_mult=0.),
nesterov=True) nesterov=True)
# learning policy # learning policy
lr_config = dict(policy='step', step=[30, 60, 90]) lr_config = dict(policy='step', step=[7, 14, 21])
checkpoint_config = dict(interval=10) checkpoint_config = dict(interval=10)
# runtime settings # runtime settings
total_epochs = 100 total_epochs = 28

8
configs/benchmarks/linear_classification/places205/r50_multihead_sobel.py
View File

@ -46,7 +46,7 @@ test_pipeline = [
] ]
data = dict( data = dict(
imgs_per_gpu=32, # total 32x8=256 imgs_per_gpu=32, # total 32x8=256
workers_per_gpu=5, workers_per_gpu=4,
train=dict( train=dict(
type=dataset_type, type=dataset_type,
data_source=dict( data_source=dict(
@ -65,7 +65,7 @@ custom_hooks = [
dataset=data['val'], dataset=data['val'],
initial=True, initial=True,
interval=10, interval=10,
imgs_per_gpu=128, imgs_per_gpu=32,
workers_per_gpu=4, workers_per_gpu=4,
eval_param=dict(topk=(1, ))) eval_param=dict(topk=(1, )))
] ]
@ -78,7 +78,7 @@ optimizer = dict(
paramwise_options=dict(norm_decay_mult=0.), paramwise_options=dict(norm_decay_mult=0.),
nesterov=True) nesterov=True)
# learning policy # learning policy
lr_config = dict(policy='step', step=[30, 60, 90]) lr_config = dict(policy='step', step=[7, 14, 21])
checkpoint_config = dict(interval=10) checkpoint_config = dict(interval=10)
# runtime settings # runtime settings
total_epochs = 100 total_epochs = 28