From 608069b766204bca929cf523668c0adc74a5d2ea Mon Sep 17 00:00:00 2001 From: xjqbest <173596896@qq.com> Date: Fri, 12 Jun 2020 14:35:46 +0800 Subject: [PATCH] fix debug --- core/trainers/framework/runner.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/core/trainers/framework/runner.py b/core/trainers/framework/runner.py index b383ff98..d11a9c0d 100644 --- a/core/trainers/framework/runner.py +++ b/core/trainers/framework/runner.py @@ -69,7 +69,8 @@ class RunnerBase(object): dataset=reader, fetch_list=fetch_vars, fetch_info=fetch_alias, - print_period=fetch_period) + print_period=fetch_period, + debug=envs.get_global_env("debug", False)) else: metrics = model_class.get_metrics() if metrics: @@ -81,7 +82,8 @@ class RunnerBase(object): dataset=reader, fetch_list=fetch_vars, fetch_info=fetch_alias, - print_period=fetch_period) + print_period=fetch_period, + debug=envs.get_global_env("debug", False)) def _executor_dataloader_train(self, model_dict, context): model_name = model_dict["name"] -- GitLab