lint

pytorch · Apr 25, 2024 · 4f77fdc · 4f77fdc
1 parent ab787b3
commit 4f77fdc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -228,7 +228,7 @@ def loss_fn(pred, labels):
             torch._dynamo.config._experimental_support_context_fn_in_torch_utils_checkpoint = (
                 True
             )
-        logger.info(f"Compiling each TransformerBlock with torch.compile")
+        logger.info("Compiling each TransformerBlock with torch.compile")
 
     train_state = TrainState()