call init later

holgerroth · Apr 26, 2024 · d6f1f58 · d6f1f58
1 parent 0145382
commit d6f1f58
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/integration/monai/examples/mednist/code/monai_mednist_train.py b/integration/monai/examples/mednist/code/monai_mednist_train.py
@@ -52,9 +52,6 @@
 
 print_config()
 
-# (2) initializes NVFlare client API
-flare.init()
-
 # Setup data directory
 directory = os.environ.get("MONAI_DATA_DIRECTORY")
 root_dir = tempfile.mkdtemp() if directory is None else directory
@@ -97,6 +94,9 @@
     train_handlers=StatsHandler(),
 )
 
+# (2) initializes NVFlare client API
+flare.init()
+
 # (optional) calculate total steps
 steps = max_epochs * len(train_loader)
 # Run the training