some debug statements

hasan7n · Oct 4, 2024 · 0fc0aa0 · 0fc0aa0
1 parent ac18c19
commit 0fc0aa0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/fl_post/fl/project/src/nnunet_v1.py b/examples/fl_post/fl/project/src/nnunet_v1.py
@@ -271,7 +271,7 @@ def __init__(self, **kwargs):
             else:
                 # new training without pretraine weights, do nothing
                 pass
-
+            print(f"Brandon DEBUG - Calling trainer.run_training, trainer epoch: {trainer.epoch}, trainer max_num_epochs:{trainer.max_num_epochs}")
             batches_applied_train, batches_applied_val = trainer.run_training(train_cutoff=train_cutoff, val_cutoff=val_cutoff, decrement_current_epoch_by_one=decrement_current_epoch_by_one)
         else:
             # if valbest: