diff --git a/core/trainers/framework/runner.py b/core/trainers/framework/runner.py index 0a1cec8421448223c2147d53635bbb22f8758e36..9064989ad467d3104adf50d64300b7903376e7ce 100644 --- a/core/trainers/framework/runner.py +++ b/core/trainers/framework/runner.py @@ -213,16 +213,12 @@ class RunnerBase(object): if context["fleet_mode"].upper() == "PS": train_prog = context["model"][model_dict[ "name"]]["main_program"] - print("condition 1 of bath id:{}".format( - batch_id)) else: train_prog = context["model"][model_dict[ "name"]]["default_main_program"] - print("condition 2") else: train_prog = context["model"][model_dict["name"]][ "default_main_program"] - print("condition 3") startup_prog = context["model"][model_dict["name"]][ "startup_program"] with fluid.program_guard(train_prog, startup_prog):