correcting training config

hasan7n · Sep 30, 2024 · c0b9f51 · c0b9f51
1 parent b59c2dc
commit c0b9f51
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/examples/fl_post/fl/mlcube/workspace/training_config.yaml b/examples/fl_post/fl/mlcube/workspace/training_config.yaml
@@ -55,11 +55,10 @@ tasks :
   defaults : plan/defaults/tasks_torch.yaml
   aggregated_model_validation:
     function : validate
-    kwargs   :
+    kwargs :
       metrics     :
-      apply  : global
       - val_eval
-      epochs : 1
+      apply : global
   train:
     function : train
     kwargs   :
@@ -72,7 +71,6 @@ tasks :
       metrics     :
       - val_eval
       apply  : local
-      epochs : 1
 
 compression_pipeline :
   defaults : plan/defaults/compression_pipeline.yaml