Skip to content

Commit 9d44598

Browse files
authored
fix: Async GRPO loop crash logging (#1321)
Signed-off-by: Bogdan Salyp <bogdansalyp@gmail.com>
1 parent 7c574d0 commit 9d44598

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

nemo_rl/algorithms/grpo.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1787,6 +1787,12 @@ def async_grpo_train(
17871787
timer.reset()
17881788
step += 1
17891789

1790+
except Exception as e:
1791+
print(f"❌ Error in async loop: {e}")
1792+
import traceback
1793+
1794+
traceback.print_exc()
1795+
17901796
finally:
17911797
# Clean up
17921798
print("🛑 Stopping trajectory collection...")

0 commit comments

Comments
 (0)