From 59d47662c2874417f13f43fe631c23e5d1a04c9f Mon Sep 17 00:00:00 2001 From: Rishi Sharma <rishi.sharma@epfl.ch> Date: Mon, 27 Dec 2021 06:01:20 +0100 Subject: [PATCH] Migrate configs --- eval/config_femnist_100.ini | 32 ++++++++++++++++++++++++++++++++ 1 file changed, 32 insertions(+) create mode 100644 eval/config_femnist_100.ini diff --git a/eval/config_femnist_100.ini b/eval/config_femnist_100.ini new file mode 100644 index 0000000..c3f0ae2 --- /dev/null +++ b/eval/config_femnist_100.ini @@ -0,0 +1,32 @@ +[DATASET] +dataset_package = decentralizepy.datasets.Femnist +dataset_class = Femnist +model_class = CNN +n_procs = 16 +train_dir = /home/risharma/leaf/data/femnist/per_user_data/train +test_dir = /home/risharma/leaf/data/femnist/data/test +; python list of fractions below +sizes = + +[OPTIMIZER_PARAMS] +optimizer_package = torch.optim +optimizer_class = Adam +lr = 0.001 + +[TRAIN_PARAMS] +training_package = decentralizepy.training.Training +training_class = Training +epochs_per_round = 5 +batch_size = 1024 +shuffle = True +loss_package = torch.nn +loss_class = CrossEntropyLoss + +[COMMUNICATION] +comm_package = decentralizepy.communication.TCP +comm_class = TCP +addresses_filepath = ip_addr_6Machines.json + +[SHARING] +sharing_package = decentralizepy.sharing.Sharing +sharing_class = Sharing -- GitLab