From 1a581da86a7cf86fb17270df3e2333c97deceace Mon Sep 17 00:00:00 2001 From: qjing666 Date: Mon, 16 Dec 2019 21:46:18 +0800 Subject: [PATCH] update --- paddle_fl/dataset/femnist.py | 2 +- paddle_fl/examples/femnist_demo/fl_trainer.py | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/paddle_fl/dataset/femnist.py b/paddle_fl/dataset/femnist.py index bac01ef..4b096c8 100644 --- a/paddle_fl/dataset/femnist.py +++ b/paddle_fl/dataset/femnist.py @@ -34,7 +34,7 @@ def train(trainer_id,inner_step,batch_size,count_by_step): rand = random.randrange(0,len(users)) # random choose a user from each trainer cur_user = users[rand] print('training using '+cur_user) - train_images = json_train["user_data"][cur_user]['x'] + train_images = json_train["user_data"][cur_user]['x'] train_labels = json_train["user_data"][cur_user]['y'] if count_by_step: for i in xrange(inner_step*batch_size): diff --git a/paddle_fl/examples/femnist_demo/fl_trainer.py b/paddle_fl/examples/femnist_demo/fl_trainer.py index 9fd28ff..9fe9886 100644 --- a/paddle_fl/examples/femnist_demo/fl_trainer.py +++ b/paddle_fl/examples/femnist_demo/fl_trainer.py @@ -61,7 +61,6 @@ while not trainer.stop(): test_reader = paddle.batch( paddle_fl.dataset.femnist.test(trainer_id,inner_step=trainer._step,batch_size=64,count_by_step=count_by_step), batch_size=64) - if count_by_step: for step_id, data in enumerate(train_reader()): acc = trainer.run(feeder.feed(data), fetch=["accuracy_0.tmp_0"]) -- GitLab