Skip to content

Commit 81e5df5

Browse files
committed
add FOR_GLOBAL_MODEL_KEY in config.utils
1 parent 275640f commit 81e5df5

1 file changed

Lines changed: 0 additions & 2 deletions

File tree

fl4health/clients/nnunet_client.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -323,8 +323,6 @@ def get_model(self, config: Config) -> nn.Module:
323323
else:
324324
return self.nnunet_trainer.network
325325

326-
return self.nnunet_trainer.network
327-
328326
def get_criterion(self, config: Config) -> _Loss:
329327
if isinstance(self.nnunet_trainer.loss, DeepSupervisionWrapper):
330328
self.reports_manager.report({"Criterion": self.nnunet_trainer.loss.loss.__class__.__name__})

0 commit comments

Comments
 (0)