Fixed torch_unet_kvasir workspace (securefederatedai#1269)

Signed-off-by: Tarunkumar Banda <[email protected]>
refai06 · Jan 16, 2025 · ad43881 · ad43881
1 parent 69a2ceb
commit ad43881
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 7 deletions.
diff --git a/openfl-workspace/torch_unet_kvasir/plan/plan.yaml b/openfl-workspace/torch_unet_kvasir/plan/plan.yaml
@@ -5,31 +5,29 @@ aggregator :
   defaults : plan/defaults/aggregator.yaml
   template : openfl.component.Aggregator
   settings :
-    init_state_path : save/torch_unet_kvasir_init.pbuf
-    best_state_path : save/torch_unet_kvasir_best.pbuf
-    last_state_path : save/torch_unet_kvasir_last.pbuf
+    init_state_path : save/init.pbuf
+    best_state_path : save/best.pbuf
+    last_state_path : save/last.pbuf
     rounds_to_train : 40
 
 collaborator :
   defaults : plan/defaults/collaborator.yaml
   template : openfl.component.Collaborator
   settings :
-    epochs_per_round : 1.0
-    polling_interval : 4
     delta_updates    : false
     opt_treatment    : RESET
 
 data_loader :
   defaults : plan/defaults/data_loader.yaml
-  template : src.data_loader.PyTorchKvasirDataLoader
+  template : src.dataloader.PyTorchKvasirDataLoader
   settings :
     collaborator_count : 2
     data_group_name    : kvasir
     batch_size         : 4
 
 task_runner :
   defaults : plan/defaults/task_runner.yaml
-  template : src.fed_unet_runner.PyTorchFederatedUnet
+  template : src.taskrunner.PyTorchFederatedUnet
   settings :
     n_channels : 3
     n_classes  : 1

diff --git a/...pace/torch_unet_kvasir/src/data_loader.py → ...space/torch_unet_kvasir/src/dataloader.py b/...pace/torch_unet_kvasir/src/data_loader.py → ...space/torch_unet_kvasir/src/dataloader.py
diff --git a/.../torch_unet_kvasir/src/fed_unet_runner.py → ...space/torch_unet_kvasir/src/taskrunner.py b/.../torch_unet_kvasir/src/fed_unet_runner.py → ...space/torch_unet_kvasir/src/taskrunner.py