config fixes

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

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