Merge pull request #826 from epfml/NAN-fix-mnist

Make mnist use mean aggregation rather than secure
epfml · Nov 7, 2024 · afe90c6 · afe90c6
2 parents a5735e8 + 0236538
commit afe90c6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/discojs/src/default_tasks/mnist.ts b/discojs/src/default_tasks/mnist.ts
@@ -30,7 +30,7 @@ export const mnist: TaskProvider<'image'> = {
         IMAGE_W: 28,
         LABEL_LIST: ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9'],
         scheme: 'decentralized',
-        aggregationStrategy: 'secure',
+        aggregationStrategy: 'mean',
         minNbOfParticipants: 3,
         maxShareValue: 100,
         tensorBackend: 'tfjs'