From b65066244cc098f38485fd8a10fa8546cd9e16c4 Mon Sep 17 00:00:00 2001 From: jingqinghe Date: Wed, 22 Apr 2020 16:30:56 +0800 Subject: [PATCH] fix bug in py3 --- paddle_fl/core/master/job_generator.py | 4 ++-- paddle_fl/examples/generate_job_from_program/program_saver.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/paddle_fl/core/master/job_generator.py b/paddle_fl/core/master/job_generator.py index e2a5e48..483aa85 100644 --- a/paddle_fl/core/master/job_generator.py +++ b/paddle_fl/core/master/job_generator.py @@ -209,12 +209,12 @@ class JobGenerator(object): program_desc_str = fin.read() new_main = fluid.Program.parse_from_string(program_desc_str) para_list = [] - with open(program_input + '/para_info', 'rb') as fin: + with open(program_input + '/para_info', 'r') as fin: for line in fin: current_para = line[:-1] para_list.append(current_para) - with open(program_input + '/loss_name', 'rb') as fin: + with open(program_input + '/loss_name', 'r') as fin: loss_name = fin.read() for item in para_list: diff --git a/paddle_fl/examples/generate_job_from_program/program_saver.py b/paddle_fl/examples/generate_job_from_program/program_saver.py index 9c609e0..742c25f 100644 --- a/paddle_fl/examples/generate_job_from_program/program_saver.py +++ b/paddle_fl/examples/generate_job_from_program/program_saver.py @@ -39,14 +39,14 @@ def save_program(program_path): para_info = [] for pa in params: para_info.append(pa.name) - with open(program_path + '/para_info', 'wb') as fout: + with open(program_path + '/para_info', 'w') as fout: for item in para_info: fout.write("%s\n" % item) with open(program_path + '/startup_program', "wb") as fout: fout.write(startup_program_str) with open(program_path + '/main_program', "wb") as fout: fout.write(main_program_str) - with open(program_path + '/loss_name', 'wb') as fout: + with open(program_path + '/loss_name', 'w') as fout: fout.write(avg_cost.name) -- GitLab