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