Update data.yaml

Correcting a mistake train dataset assigned to test and vice versa. Signed-off-by: Parth Mandaliya <parthx.mandaliya@intel.com>
ParthMandaliya · Sep 28, 2023 · 07f283f · 07f283f
1 parent 265bf82
commit 07f283f
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/openfl-workspace/experimental/101_torch_cnn_mnist/plan/data.yaml b/openfl-workspace/experimental/101_torch_cnn_mnist/plan/data.yaml
@@ -12,8 +12,8 @@ col1:
       batch_size: 32
       index: 0
       n_collaborators: 2
-      train_dataset: src.collaborator_private_attrs.test_dataset
-      test_dataset: src.collaborator_private_attrs.train_dataset
+      train_dataset: src.collaborator_private_attrs.train_dataset
+      test_dataset: src.collaborator_private_attrs.test_dataset
     template: src.collaborator_private_attrs.collaborator_private_attrs
 
 col2:
@@ -22,6 +22,6 @@ col2:
       batch_size: 32
       index: 1
       n_collaborators: 2
-      train_dataset: src.collaborator_private_attrs.test_dataset
-      test_dataset: src.collaborator_private_attrs.train_dataset
+      train_dataset: src.collaborator_private_attrs.train_dataset
+      test_dataset: src.collaborator_private_attrs.test_dataset
     template: src.collaborator_private_attrs.collaborator_private_attrs