|
|
|
@ -7,8 +7,17 @@ width = 227
|
|
|
|
|
num_class = 1000
|
|
|
|
|
batch_size = get_config_arg('batch_size', int, 128)
|
|
|
|
|
use_mkldnn = get_config_arg('use_mkldnn', bool, False)
|
|
|
|
|
is_infer = get_config_arg("is_infer", bool, False)
|
|
|
|
|
num_samples = get_config_arg('num_samples', int, 2560)
|
|
|
|
|
|
|
|
|
|
args = {'height': height, 'width': width, 'color': True, 'num_class': num_class}
|
|
|
|
|
args = {
|
|
|
|
|
'height': height,
|
|
|
|
|
'width': width,
|
|
|
|
|
'color': True,
|
|
|
|
|
'num_class': num_class,
|
|
|
|
|
'is_infer': is_infer,
|
|
|
|
|
'num_samples': num_samples
|
|
|
|
|
}
|
|
|
|
|
define_py_data_sources2(
|
|
|
|
|
"train.list", None, module="provider", obj="process", args=args)
|
|
|
|
|
|
|
|
|
|