add init log

ananthsub · ananthsub · commit a747d1c71ebd · 2026-01-28T17:41:08.000-08:00
Signed-off-by: Ananth Subramaniam &lt;ansubramania@nvidia.com&gt;
diff --git a/src/megatron/bridge/training/nvrx_straggler.py b/src/megatron/bridge/training/nvrx_straggler.py
@@ -88,7 +88,7 @@ def initialize(self) -> None:
         )
 
         self.initialized = True
-        self.logger.debug("NVRx straggler detection initialized successfully.")
+        self.logger.info(f"NVRx straggler detection enabled (report_interval={self.config.report_time_interval}s)")
 
     def wrap_train_step_function(self, train_step_func: Callable) -> Callable:
         """

Original file line number	Diff line number	Diff line change
`@@ -88,7 +88,7 @@ def initialize(self) -> None:`
`88`	`88`	`)`
`89`	`89`
`90`	`90`	`self.initialized = True`
`91`		`- self.logger.debug("NVRx straggler detection initialized successfully.")`
	`91`	`+ self.logger.info(f"NVRx straggler detection enabled (report_interval={self.config.report_time_interval}s)")`
`92`	`92`
`93`	`93`	`def wrap_train_step_function(self, train_step_func: Callable) -> Callable:`
`94`	`94`	`"""`