add training mode

pytorch · Feb 1, 2025 · 96cee89 · 96cee89
1 parent e9f372e
commit 96cee89
Show file tree

Hide file tree

Showing 2 changed files with 283 additions and 241 deletions.
diff --git a/debugging/repros/attention.py b/debugging/repros/attention.py