From ba2a2787aac1ff75219171b2bc0d154ba8bebf65 Mon Sep 17 00:00:00 2001
From: UltralyticsAssistant <web@ultralytics.com>
Date: Sun, 5 Jan 2025 16:00:08 +0000
Subject: [PATCH] Auto-format by https://ultralytics.com

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

diff --git a/train.py b/train.py
index 6ad263722bbf..6c967bc6fa8e 100644
--- a/train.py
+++ b/train.py
@@ -37,7 +37,6 @@
 import yaml
 from torch.optim import lr_scheduler
 from tqdm import tqdm
-from torch.cuda import amp  
 
 FILE = Path(__file__).resolve()
 ROOT = FILE.parents[0]  # YOLOv5 root directory