From 27f9f785bdaf6d5f9e811191cfbd1b027b7bcf11 Mon Sep 17 00:00:00 2001
From: Jerseyshin <50386604+Jerseyshin@users.noreply.github.com>
Date: Mon, 24 Jun 2024 16:42:27 +0800
Subject: [PATCH] Update trainer_worker.py

---
 fedlearner/trainer/trainer_worker.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/fedlearner/trainer/trainer_worker.py b/fedlearner/trainer/trainer_worker.py
index caa5bf292..1fc8c51c0 100644
--- a/fedlearner/trainer/trainer_worker.py
+++ b/fedlearner/trainer/trainer_worker.py
@@ -202,6 +202,18 @@ def create_argument_parser():
                         type=str,
                         default='0.25',
                         help='Marvell sumKL threshold.')
+    parser.add_argument('--using_fedpass',
+                        type=str_as_bool,
+                        default='False',
+                        help='Whether use fedpass protection.')
+    parser.add_argument('--fedpass_mean',
+                        type=str,
+                        default='50.0',
+                        help='FedPass secretkey mean.')
+    parser.add_argument('--fedpass_scale',
+                        type=str,
+                        default='5.0',
+                        help='FedPass secretkey scale.')
     parser.add_argument('--using_emb_attack',
                         type=str_as_bool,
                         default='False',