Update train.py

improve the print message
scaleoutsystems · May 21, 2024 · 81ee6ee · 81ee6ee
1 parent f2f7b51
commit 81ee6ee
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/monai-2D-mednist/client/train.py b/examples/monai-2D-mednist/client/train.py
@@ -137,7 +137,7 @@ def train(in_model_path, out_model_path, data_path=None, client_settings_path=No
         epoch_loss_values.append(epoch_loss)
         print(f"epoch {epoch + 1} average loss: {epoch_loss:.4f}")
 
-    print(f"train completed!")
+    print(f"training completed!")
 
     # Metadata needed for aggregation server side
     metadata = {