Skip to content

Commit 5830978

Browse files
committed
Warning LF with FedAvg not supported
1 parent 69bd1ac commit 5830978

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

byzfl/benchmark/train.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -206,7 +206,9 @@ def start_training(params):
206206
if training_algorithm_name not in ["DSGD", "FedAvg"]:
207207
raise ValueError(f"Training algorithm {training_algorithm_name} not supported, supported algorithms are 'DSGD' and 'FedAvg'")
208208

209-
209+
if training_algorithm_name == "FedAvg" and attack_name == "LabelFlipping":
210+
raise ValueError("FedAvg does not support Label Flipping attack.")
211+
210212
if training_algorithm_name == "FedAvg":
211213

212214
training_algorithm_parameters = params_manager.get_training_algorithm_parameters()

0 commit comments

Comments
 (0)