From 317cf8733d1b30b899c89e2c4d46c2a63e128bbe Mon Sep 17 00:00:00 2001 From: "Edwards, Brandon" Date: Mon, 28 Oct 2024 10:49:24 -0700 Subject: [PATCH] doc string --- examples/fl_post/fl/project/src/nnunet_v1.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/fl_post/fl/project/src/nnunet_v1.py b/examples/fl_post/fl/project/src/nnunet_v1.py index e539a8913..bdf6a0393 100644 --- a/examples/fl_post/fl/project/src/nnunet_v1.py +++ b/examples/fl_post/fl/project/src/nnunet_v1.py @@ -84,9 +84,9 @@ def train_nnunet(actual_max_num_epochs, """ actual_max_num_epochs (int): Provides the number of epochs intended to be trained (this needs to be held constant outside of individual calls to this function during with max_num_epochs is set to one more than the current epoch) - fl_round (int): Federated round, equal to the epoch used for the model (lr scheduling) + fl_round (int): Federated round, equal to the epoch used for the model (in lr scheduling) val_epoch (bool) : Will validation be performed - train_epoch (bool) : Will training run (rather than val only) makes lr step and epoch increment + train_epoch (bool) : Will training run (rather than val only) train_val_cutoff (int): Total time (in seconds) limit to use in approximating a restriction to training and validation activities. train_cutoff_part (float): Portion of train_val_cutoff going to training val_cutoff_part (float): Portion of train_val_cutoff going to val