Updated 101_torch_cnn_mnist private_attrs

Signed-off-by: Ishant Thakare <[email protected]>
securefederatedai · May 31, 2024 · a920ecf · a920ecf
1 parent 3632eb6
commit a920ecf
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/openfl-workspace/experimental/101_torch_cnn_mnist/src/collaborator_private_attrs.py b/openfl-workspace/experimental/101_torch_cnn_mnist/src/collaborator_private_attrs.py
@@ -38,10 +38,10 @@
 train = deepcopy(mnist_train)
 test = deepcopy(mnist_test)
 
-train.data = train.data[0::n_collaborators]
-train.targets = train.targets[0::n_collaborators]
-test.data = test.data[0::n_collaborators]
-test.targets = test.targets[0::n_collaborators]
+train.data = mnist_train.data[0::n_collaborators]
+train.targets = mnist_train.targets[0::n_collaborators]
+test.data = mnist_test.data[0::n_collaborators]
+test.targets = mnist_test.targets[0::n_collaborators]
 
 col1_private_attributes = {
     "train_loader": torch.utils.data.DataLoader(
@@ -52,10 +52,10 @@
     ),
 }
 
-train.data = train.data[1::n_collaborators]
-train.targets = train.targets[1::n_collaborators]
-test.data = test.data[1::n_collaborators]
-test.targets = test.targets[1::n_collaborators]
+train.data = mnist_train.data[1::n_collaborators]
+train.targets = mnist_train.targets[1::n_collaborators]
+test.data = mnist_test.data[1::n_collaborators]
+test.targets = mnist_test.targets[1::n_collaborators]
 
 col2_private_attributes = {
     "train_loader": torch.utils.data.DataLoader(