Skip to content

Commit 3ea7f1c

Browse files
committed
Update start_server.py
Change the default dataset to MNIST
1 parent fe5f0b3 commit 3ea7f1c

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

federa/server/start_server.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@
2929
parser.add_argument('--resize_size', type = int, default = 32, help= 'resize dimension')
3030
parser.add_argument('--batch_size', type = int, default = 32, help= 'batch size')
3131
parser.add_argument('--net', type = str, default = 'LeNet', help= 'client network')
32-
parser.add_argument('--dataset', type = str, default= 'FashionMNIST',
32+
parser.add_argument('--dataset', type = str, default= 'MNIST',
3333
help= 'datsset.Use CUSTOME for local dataset')
3434
parser.add_argument('--niid', type = int, default= 1, help= 'value should be [1, 5]')
3535
parser.add_argument('--carbon', type = int, default= 0,

0 commit comments

Comments
 (0)