From bc7e3331f188ef3c0384a70d3ec296ebd4b26ec9 Mon Sep 17 00:00:00 2001
From: UltralyticsAssistant <web@ultralytics.com>
Date: Wed, 22 Jan 2025 16:37:28 +0000
Subject: [PATCH] Auto-format by https://ultralytics.com

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index babbc4c9f..fa04089d4 100644
--- a/train.py
+++ b/train.py
@@ -409,7 +409,7 @@ def train(hyp, opt, device, callbacks):
                     imgs = nn.functional.interpolate(imgs, size=ns, mode="bilinear", align_corners=False)
 
             # Forward
-            with torch.amp.autocast('cuda', amp):
+            with torch.amp.autocast("cuda", amp):
                 pred = model(imgs)  # forward
                 loss, loss_items = compute_loss(pred, targets.to(device))  # loss scaled by batch_size
                 if RANK != -1: