Skip to content

Commit 0c53981

Browse files
committed
[debug] fix standalone example
1 parent edc47f4 commit 0c53981

File tree

4 files changed

+4
-2
lines changed

4 files changed

+4
-2
lines changed

.gitignore

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -199,3 +199,4 @@ unit_test*
199199
/datasets/*/meta/*
200200
/datasets/*/train/*
201201
/datasets/*/test/*
202+
/datasets/*/*

examples/standalone-mnist/standalone.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,11 +38,11 @@
3838

3939
# server
4040
handler = SyncServerHandler(
41-
model, args.com_round, args.total_clients, args.sample_ratio
41+
model, args.com_round, args.sample_ratio
4242
)
4343

4444
# client
45-
trainer = SGDSerialClientTrainer(model, args.total_clients, cuda=True)
45+
trainer = SGDSerialClientTrainer(model, args.total_clients, cuda=False)
4646
dataset = PathologicalMNIST(
4747
root="../../datasets/mnist/",
4848
path="../../datasets/mnist/",
@@ -53,6 +53,7 @@
5353
trainer.setup_dataset(dataset)
5454
trainer.setup_optim(args.epochs, args.batch_size, args.lr)
5555

56+
handler.num_clients = args.total_clients
5657
handler.setup_dataset(dataset)
5758
# main
5859
pipeline = StandalonePipeline(handler, trainer)

fedlab/contrib/algorithm/fedasync.py

Whitespace-only changes.

fedlab/contrib/algorithm/fedbuff.py

Whitespace-only changes.

0 commit comments

Comments
 (0)