diff --git a/models/multitask/esmm/config.yaml b/models/multitask/esmm/config.yaml index 01c8fa61029e9eab1f4f594e0f1735df5737e6dd..9606ca79320a20bd27175c5ce91c1c1c7f3513fe 100644 --- a/models/multitask/esmm/config.yaml +++ b/models/multitask/esmm/config.yaml @@ -37,7 +37,6 @@ hyper_parameters: #use infer_runner mode and modify 'phase' below if infer mode: [train_runner, infer_runner] -#mode: infer_runner runner: - name: train_runner @@ -55,6 +54,7 @@ runner: init_model_path: "increment/0" device: cpu epochs: 1 + print_interval: 1 phases: [infer] phase: