Skip to content
Snippets Groups Projects
Commit 1a56aadc authored by Jeffrey Wigger's avatar Jeffrey Wigger
Browse files

config fixes

parent 90772eb9
No related branches found
No related tags found
No related merge requests found
......@@ -3,8 +3,8 @@ dataset_package = decentralizepy.datasets.Femnist
dataset_class = Femnist
random_seed = 97
model_class = CNN
train_dir = /home/risharma/leaf/data/femnist/per_user_data/train
test_dir = /home/risharma/leaf/data/femnist/data/test
train_dir = /mnt/nfs/shared/leaf/data/femnist/per_user_data/train
test_dir = /mnt/nfs/shared/leaf/data/femnist/data/test
; python list of fractions below
sizes =
......@@ -16,9 +16,9 @@ lr = 0.001
[TRAIN_PARAMS]
training_package = decentralizepy.training.GradientAccumulator
training_class = GradientAccumulator
rounds = 20
rounds = 47
full_epochs = False
batch_size = 64
batch_size = 16
shuffle = True
loss_package = torch.nn
loss_class = CrossEntropyLoss
......
......@@ -31,6 +31,6 @@ comm_class = TCP
addresses_filepath = ip_addr_6Machines.json
[SHARING]
sharing_package = decentralizepy.sharing.TopK
sharing_class = TopK
sharing_package = decentralizepy.sharing.PartialModel
sharing_class = PartialModel
alpha = 0.1
\ No newline at end of file
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment