From 251e6848406d92b46fa4cad64968d5c6e25da0b7 Mon Sep 17 00:00:00 2001 From: shehper Date: Tue, 9 Apr 2024 07:56:04 -0400 Subject: [PATCH] suppressing a seemingly bening error that seems to appear with TorchDynamo in transformer/train.py --- transformer/train.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/transformer/train.py b/transformer/train.py index a482ab7..b9e07e2 100644 --- 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