From 64dc0f63c34b3852fd730ae24aa875b23e586204 Mon Sep 17 00:00:00 2001
From: Jeffrey Wigger <jeffrey.wigger@epfl.ch>
Date: Thu, 17 Feb 2022 13:16:51 +0100
Subject: [PATCH] using uid instead of rank for the femnist data partition
---
src/decentralizepy/datasets/Femnist.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/src/decentralizepy/datasets/Femnist.py b/src/decentralizepy/datasets/Femnist.py
index c0360c5..2e80096 100644
--- a/src/decentralizepy/datasets/Femnist.py
+++ b/src/decentralizepy/datasets/Femnist.py
@@ -123,7 +123,8 @@ class Femnist(Dataset):
self.sizes[-1] += 1.0 - frac * self.n_procs
logging.debug("Size fractions: {}".format(self.sizes))
- my_clients = DataPartitioner(files, self.sizes).use(self.rank)
+ self.uid = self.mapping.get_uid(self.rank, self.machine_id)
+ my_clients = DataPartitioner(files, self.sizes).use(self.uid)
my_train_data = {"x": [], "y": []}
self.clients = []
self.num_samples = []
--
GitLab