diff --git a/openfl-workspace/torch_unet_kvasir/plan/plan.yaml b/openfl-workspace/torch_unet_kvasir/plan/plan.yaml index b37a8a7d91..6563a31075 100644 --- a/openfl-workspace/torch_unet_kvasir/plan/plan.yaml +++ b/openfl-workspace/torch_unet_kvasir/plan/plan.yaml @@ -5,23 +5,21 @@ 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 @@ -29,7 +27,7 @@ data_loader : 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/openfl-workspace/torch_unet_kvasir/src/data_loader.py b/openfl-workspace/torch_unet_kvasir/src/dataloader.py similarity index 100% rename from openfl-workspace/torch_unet_kvasir/src/data_loader.py rename to openfl-workspace/torch_unet_kvasir/src/dataloader.py diff --git a/openfl-workspace/torch_unet_kvasir/src/fed_unet_runner.py b/openfl-workspace/torch_unet_kvasir/src/taskrunner.py similarity index 100% rename from openfl-workspace/torch_unet_kvasir/src/fed_unet_runner.py rename to openfl-workspace/torch_unet_kvasir/src/taskrunner.py