suppressing a seemingly bening error that seems to appear with TorchD…

…ynamo in transformer/train.py
shehper · Apr 9, 2024 · 251e684 · 251e684
1 parent e262cb4
commit 251e684
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/transformer/train.py b/transformer/train.py
@@ -29,6 +29,10 @@
 
 from model import GPTConfig, GPT
 
+
+import torch._dynamo
+torch._dynamo.config.suppress_errors = True
+
 # -----------------------------------------------------------------------------
 # default config values designed to train a gpt2 (124M) on OpenWebText
 # I/O