From 7bca046d0d78b1cd6cbd03b0ac06ef89d269cb2c Mon Sep 17 00:00:00 2001 From: zhoushan Date: Fri, 25 Feb 2022 17:15:44 +0800 Subject: [PATCH] change lr for server in master --- tests/st/fl/albert/run_hybrid_train_server.py | 2 +- tests/st/fl/cross_device_lenet/cloud/run_lenet_server.py | 2 +- tests/st/fl/mobile/run_mobile_server.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/st/fl/albert/run_hybrid_train_server.py b/tests/st/fl/albert/run_hybrid_train_server.py index e19827f3170..4b06b722d6c 100644 --- a/tests/st/fl/albert/run_hybrid_train_server.py +++ b/tests/st/fl/albert/run_hybrid_train_server.py @@ -34,7 +34,7 @@ parser.add_argument("--fl_name", type=str, default="Lenet") parser.add_argument("--fl_iteration_num", type=int, default=25) parser.add_argument("--client_epoch_num", type=int, default=20) parser.add_argument("--client_batch_size", type=int, default=32) -parser.add_argument("--client_learning_rate", type=float, default=0.1) +parser.add_argument("--client_learning_rate", type=float, default=1e-5) # The number of servers that this script will launch. parser.add_argument("--local_server_num", type=int, default=-1) parser.add_argument("--encrypt_type", type=str, default="NOT_ENCRYPT") diff --git a/tests/st/fl/cross_device_lenet/cloud/run_lenet_server.py b/tests/st/fl/cross_device_lenet/cloud/run_lenet_server.py index f03fe2e19f9..6283fdb8583 100644 --- a/tests/st/fl/cross_device_lenet/cloud/run_lenet_server.py +++ b/tests/st/fl/cross_device_lenet/cloud/run_lenet_server.py @@ -34,7 +34,7 @@ parser.add_argument("--fl_name", type=str, default="Lenet") parser.add_argument("--fl_iteration_num", type=int, default=25) parser.add_argument("--client_epoch_num", type=int, default=20) parser.add_argument("--client_batch_size", type=int, default=32) -parser.add_argument("--client_learning_rate", type=float, default=0.1) +parser.add_argument("--client_learning_rate", type=float, default=0.01) parser.add_argument("--local_server_num", type=int, default=-1) parser.add_argument("--config_file_path", type=str, default="") parser.add_argument("--encrypt_type", type=str, default="NOT_ENCRYPT") diff --git a/tests/st/fl/mobile/run_mobile_server.py b/tests/st/fl/mobile/run_mobile_server.py index 8d2cb049c01..8073757b387 100644 --- a/tests/st/fl/mobile/run_mobile_server.py +++ b/tests/st/fl/mobile/run_mobile_server.py @@ -34,7 +34,7 @@ parser.add_argument("--fl_name", type=str, default="Lenet") parser.add_argument("--fl_iteration_num", type=int, default=25) parser.add_argument("--client_epoch_num", type=int, default=20) parser.add_argument("--client_batch_size", type=int, default=32) -parser.add_argument("--client_learning_rate", type=float, default=0.1) +parser.add_argument("--client_learning_rate", type=float, default=0.01) parser.add_argument("--local_server_num", type=int, default=-1) parser.add_argument("--config_file_path", type=str, default="") parser.add_argument("--pki_verify", type=ast.literal_eval, default=False)