From c4b5ef450e3ad17512fa3651228f3040ce61e839 Mon Sep 17 00:00:00 2001 From: Kevin Ta <116312994+kta-intel@users.noreply.github.com> Date: Mon, 27 Nov 2023 16:16:08 -0500 Subject: [PATCH] Add comments to plan.yaml Signed-off-by: kta-intel --- openfl-workspace/torch_cnn_mnist_fed_eval/plan/plan.yaml | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/openfl-workspace/torch_cnn_mnist_fed_eval/plan/plan.yaml b/openfl-workspace/torch_cnn_mnist_fed_eval/plan/plan.yaml index 989c5af94e..2eca7b67c4 100644 --- a/openfl-workspace/torch_cnn_mnist_fed_eval/plan/plan.yaml +++ b/openfl-workspace/torch_cnn_mnist_fed_eval/plan/plan.yaml @@ -8,7 +8,9 @@ aggregator : init_state_path : save/torch_cnn_mnist_init.pbuf best_state_path : save/torch_cnn_mnist_best.pbuf last_state_path : save/torch_cnn_mnist_last.pbuf + ######### SET ROUNDS TO 1 ############# rounds_to_train : 1 + ####################################### log_metric_callback : template : src.mnist_utils.write_metric @@ -35,6 +37,7 @@ network : defaults : plan/defaults/network.yaml assigner : + ######### SET ASSIGNER TO ONLY INCLUDE AGGREGATED MODEL VALIDATION ############# template : openfl.component.RandomGroupedAssigner settings : task_groups : @@ -42,14 +45,17 @@ assigner : percentage : 1.0 tasks : - aggregated_model_validation + ################################################################################ tasks : + ######### SET AGGREGATED MODEL VALIDATION AS ONLY TASK ############# aggregated_model_validation: function : validate kwargs : apply : global metrics : - acc + #################################################################### compression_pipeline : - defaults : plan/defaults/compression_pipeline.yaml \ No newline at end of file + defaults : plan/defaults/compression_pipeline.yaml