From 1a56aadc4a579595db0b6db4353ce760350421c6 Mon Sep 17 00:00:00 2001
From: Jeffrey Wigger <jeffrey.wigger@epfl.ch>
Date: Mon, 14 Mar 2022 20:11:15 +0100
Subject: [PATCH] config fixes

---
 eval/step_configs/config_femnist.ini         | 8 ++++----
 eval/step_configs/config_femnist_topkacc.ini | 4 ++--
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/eval/step_configs/config_femnist.ini b/eval/step_configs/config_femnist.ini
index fa49fc9..4814b8a 100644
--- a/eval/step_configs/config_femnist.ini
+++ b/eval/step_configs/config_femnist.ini
@@ -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
diff --git a/eval/step_configs/config_femnist_topkacc.ini b/eval/step_configs/config_femnist_topkacc.ini
index e65f225..805004b 100644
--- a/eval/step_configs/config_femnist_topkacc.ini
+++ b/eval/step_configs/config_femnist_topkacc.ini
@@ -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
-- 
GitLab