diff --git a/core/trainers/single_infer.py b/core/trainers/single_infer.py index 873ff010416a4a3eecb88edb63dbb9c2adbf27da..0f1c92f3f2d948c76d4cd2b0fdcca131b99cfc92 100755 --- a/core/trainers/single_infer.py +++ b/core/trainers/single_infer.py @@ -229,7 +229,7 @@ class SingleInfer(TranspileTrainer): fetch_alias = [] fetch_period = int( envs.get_global_env("runner." + self._runner_name + - ".fetch_period", 20)) + ".print_interval", 20)) metrics = model_class.get_infer_results() if metrics: fetch_vars = metrics.values() @@ -260,7 +260,7 @@ class SingleInfer(TranspileTrainer): metrics_format = [] fetch_period = int( envs.get_global_env("runner." + self._runner_name + - ".fetch_period", 20)) + ".print_interval", 20)) metrics_format.append("{}: {{}}".format("batch")) for name, var in metrics.items(): metrics_varnames.append(var.name) diff --git a/core/trainers/single_trainer.py b/core/trainers/single_trainer.py index 274047a0127d6ef2df601284cb9cd1e67335cf85..66daf40276c0230505c00baa5ddf79449198074c 100755 --- a/core/trainers/single_trainer.py +++ b/core/trainers/single_trainer.py @@ -226,7 +226,7 @@ class SingleTrainer(TranspileTrainer): fetch_alias = [] fetch_period = int( envs.get_global_env("runner." + self._runner_name + - ".fetch_period", 20)) + ".print_interval", 20)) metrics = model_class.get_metrics() if metrics: fetch_vars = metrics.values() @@ -253,7 +253,7 @@ class SingleTrainer(TranspileTrainer): fetch_alias = [] fetch_period = int( envs.get_global_env("runner." + self._runner_name + - ".fetch_period", 20)) + ".print_interval", 20)) metrics = model_class.get_metrics() if metrics: fetch_vars = metrics.values() diff --git a/models/rank/dnn/config.yaml b/models/rank/dnn/config.yaml index 57bb81d56721ff875c8a9a747ed56ac100582ec2..fd64935dd2080291fc13911befc0481604c3464a 100755 --- a/models/rank/dnn/config.yaml +++ b/models/rank/dnn/config.yaml @@ -62,7 +62,7 @@ runner: save_inference_feed_varnames: [] # feed vars of save inference save_inference_fetch_varnames: [] # fetch vars of save inference init_model_path: "" # load model path - fetch_period: 10 + print_interval: 10 - name: runner2 class: single_infer # num of epochs