104 lines
5.5 KiB
Python
104 lines
5.5 KiB
Python
from collections import OrderedDict
|
|
|
|
|
|
|
|
|
|
primitives_1 = OrderedDict([('primitives_normal', [['skip_connect',
|
|
'dil_conv_3x3'],
|
|
['skip_connect',
|
|
'dil_conv_5x5'],
|
|
['skip_connect',
|
|
'dil_conv_5x5'],
|
|
['skip_connect',
|
|
'sep_conv_3x3'],
|
|
['skip_connect',
|
|
'dil_conv_3x3'],
|
|
['max_pool_3x3',
|
|
'skip_connect'],
|
|
['skip_connect',
|
|
'sep_conv_3x3'],
|
|
['skip_connect',
|
|
'sep_conv_3x3'],
|
|
['skip_connect',
|
|
'dil_conv_3x3'],
|
|
['skip_connect',
|
|
'sep_conv_3x3'],
|
|
['max_pool_3x3',
|
|
'skip_connect'],
|
|
['skip_connect',
|
|
'dil_conv_3x3'],
|
|
['dil_conv_3x3',
|
|
'dil_conv_5x5'],
|
|
['dil_conv_3x3',
|
|
'dil_conv_5x5']]),
|
|
('primitives_reduct', [['max_pool_3x3',
|
|
'avg_pool_3x3'],
|
|
['max_pool_3x3',
|
|
'dil_conv_3x3'],
|
|
['max_pool_3x3',
|
|
'avg_pool_3x3'],
|
|
['max_pool_3x3',
|
|
'avg_pool_3x3'],
|
|
['skip_connect',
|
|
'dil_conv_5x5'],
|
|
['max_pool_3x3',
|
|
'avg_pool_3x3'],
|
|
['max_pool_3x3',
|
|
'sep_conv_3x3'],
|
|
['skip_connect',
|
|
'dil_conv_3x3'],
|
|
['skip_connect',
|
|
'dil_conv_5x5'],
|
|
['max_pool_3x3',
|
|
'avg_pool_3x3'],
|
|
['max_pool_3x3',
|
|
'avg_pool_3x3'],
|
|
['skip_connect',
|
|
'dil_conv_5x5'],
|
|
['skip_connect',
|
|
'dil_conv_5x5'],
|
|
['skip_connect',
|
|
'dil_conv_5x5']])])
|
|
|
|
primitives_2 = OrderedDict([('primitives_normal', 14 * [['skip_connect',
|
|
'sep_conv_3x3']]),
|
|
('primitives_reduct', 14 * [['skip_connect',
|
|
'sep_conv_3x3']])])
|
|
|
|
primitives_3 = OrderedDict([('primitives_normal', 14 * [['none',
|
|
'skip_connect',
|
|
'sep_conv_3x3']]),
|
|
('primitives_reduct', 14 * [['none',
|
|
'skip_connect',
|
|
'sep_conv_3x3']])])
|
|
|
|
primitives_4 = OrderedDict([('primitives_normal', 14 * [['noise',
|
|
'sep_conv_3x3']]),
|
|
('primitives_reduct', 14 * [['noise',
|
|
'sep_conv_3x3']])])
|
|
|
|
PRIMITIVES = [
|
|
#'none', #0
|
|
'max_pool_3x3', # 0
|
|
'avg_pool_3x3', # 1
|
|
'skip_connect', # 2
|
|
'sep_conv_3x3', # 3
|
|
'sep_conv_5x5', # 4
|
|
'dil_conv_3x3', # 5
|
|
'dil_conv_5x5' # 6
|
|
]
|
|
|
|
primitives_5 = OrderedDict([('primitives_normal', 14 * [PRIMITIVES]),
|
|
('primitives_reduct', 14 * [PRIMITIVES])])
|
|
|
|
primitives_6 = OrderedDict([('primitives_normal', 14 * [['sep_conv_5x5']]),
|
|
('primitives_reduct', 14 * [['sep_conv_5x5']])])
|
|
spaces_dict = {
|
|
's1': primitives_1,
|
|
's2': primitives_2,
|
|
's3': primitives_3,
|
|
's4': primitives_4,
|
|
's5': primitives_5, # DARTS Space
|
|
's6': primitives_6,
|
|
}
|