[perf] Reduce job metrics reporting sleep interval to 15 seconds and …

…disable MQTT debug logging
FedML-AI · Feb 18, 2025 · 480ca1f · 480ca1f
1 parent da84c87
commit 480ca1f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/python/fedml/core/distributed/communication/mqtt/mqtt_manager.py b/python/fedml/core/distributed/communication/mqtt/mqtt_manager.py
@@ -96,9 +96,9 @@ def loop_forever(self):
         self._client.loop_forever(retry_first_connection=True)
 
     def send_message(self, topic, message, publish_single_message=False):
-        logging.info(
-            f"FedMLDebug - Send: topic ({topic}), message ({message})"
-        )
+        # logging.info(
+        #     f"FedMLDebug - Send: topic ({topic}), message ({message})"
+        # )
         self.check_connection()
 
         mqtt_send_start_time = time.time()
@@ -116,9 +116,9 @@ def send_message(self, topic, message, publish_single_message=False):
         return 0
 
     def send_message_json(self, topic, message, publish_single_message=False):
-        logging.info(
-            f"FedMLDebug - Send: topic ({topic}), message ({message})"
-        )
+        # logging.info(
+        #     f"FedMLDebug - Send: topic ({topic}), message ({message})"
+        # )
         self.check_connection()
 
         if publish_single_message:

diff --git a/python/fedml/core/mlops/mlops_job_perfs.py b/python/fedml/core/mlops/mlops_job_perfs.py
@@ -175,7 +175,7 @@ def report_job_stats_entry(self, sys_event):
                     logging.debug("exception when reporting job pref: {}.".format(traceback.format_exc()))
                     pass
 
-            time.sleep(20)
+            time.sleep(15)
 
         logging.info("Job metrics process is about to exit.")
         mqtt_mgr.loop_stop()