提交 4429973f 编写于 作者: L LDOUBLEV

revert profile_options

上级 e9e8f409
......@@ -10,7 +10,7 @@ Train.loader.batch_size_per_card:benchmark_train=16
Global.pretrained_model:null
train_model_name:latest
train_infer_img_dir:null
--profiler_options:True
--profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile
##
trainer:norm_train
norm_train:tools/train.py -c configs/det/det_mv3_db.yml -o Global.pretrained_model=./pretrain_models/MobileNetV3_large_x0_5_pretrained
......
......@@ -46,8 +46,8 @@ class ArgsParser(ArgumentParser):
self.add_argument(
'-p',
'--profiler_options',
type=bool,
default=False,
type=str,
default=None,
help='The option of profiler, which should be in format \"key1=value1;key2=value2;key3=value3\".'
)
......@@ -150,10 +150,6 @@ def train(config,
print_batch_step = config['Global']['print_batch_step']
eval_batch_step = config['Global']['eval_batch_step']
profiler_options = config['profiler_options']
if profiler_options is True:
profiler_options = "batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile"
else:
profiler_options = None
global_step = 0
if 'global_step' in pre_best_model_dict:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册