Auto-format by https://ultralytics.com

ultralytics · Jan 5, 2025 · ba2a278 · ba2a278
1 parent df838a4
commit ba2a278
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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