Change RuntimeError to print

Signed-off-by: Jaemin Choi <[email protected]>
NVIDIA · May 2, 2024 · 76d9c82 · 76d9c82
1 parent ad21c9a
commit 76d9c82
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/nemo/core/optim/distributed_adam.py b/nemo/core/optim/distributed_adam.py
@@ -171,9 +171,7 @@ def lock_with_timeout():
                         self._lock.release()
                     else:
                         # Failed to acquire lock before timeout
-                        raise RuntimeError(
-                            f'MegatronDistributedFusedAdam: Failed to acquire lock within {lock_timeout} seconds.'
-                        )
+                        print(f'MegatronDistributedFusedAdam: Failed to acquire lock within {lock_timeout} seconds.')
 
             self._lock_with_timeout = lock_with_timeout