diff --git a/eval/step_configs/config_femnist.ini b/eval/step_configs/config_femnist.ini
index fa49fc9307e9c735c5743791255002006fceed8f..4814b8a3e077b06ad1597cefa0a66683f3d9d496 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 e65f225d392716bdcbd1fcd90e52a6de23716c52..805004b6d56c95b3e79cd34debc0bc579e645ca5 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