eval/step_configs/config_celeba_grow.ini

@@ -2,24 +2,23 @@
 dataset_package = decentralizepy.datasets.Celeba
 dataset_class = Celeba
 model_class = CNN
-n_procs = 96
-images_dir = /home/risharma/leaf/data/celeba/data/raw/img_align_celeba
-train_dir = /home/risharma/leaf/data/celeba/per_user_data/train
-test_dir = /home/risharma/leaf/data/celeba/data/test
+images_dir = /mnt/nfs/shared/leaf/data/celeba/data/raw/img_align_celeba
+train_dir = /mnt/nfs/shared/leaf/data/celeba/per_user_data/train
+test_dir = /mnt/nfs/shared/leaf/data/celeba/data/test
 ; python list of fractions below
 sizes = 
 
 [OPTIMIZER_PARAMS]
 optimizer_package = torch.optim
-optimizer_class = Adam
+optimizer_class = SGD
 lr = 0.001
 
 [TRAIN_PARAMS]
-training_package = decentralizepy.training.GradientAccumulator
-training_class = GradientAccumulator
-rounds = 20
+training_package = decentralizepy.training.Training
+training_class = Training
+rounds = 4
 full_epochs = False
-batch_size = 64
+batch_size = 16
 shuffle = True
 loss_package = torch.nn
 loss_class = CrossEntropyLoss