Auto-format by https://ultralytics.com

2025-01-08 14:58:28 +00:00 · 2025-01-08 14:58:28 +00:00 · d4c977fb42
parent 1fbde181d4
commit d4c977fb42
1 changed files with 1 additions and 1 deletions
--- a/classify/train.py
+++ b/classify/train.py
@ -218,7 +218,7 @@ def train(opt, device):
            images, labels = images.to(device, non_blocking=True), labels.to(device)

            # Forward
-            with torch.amp.autocast("cuda", enabled=device.type != "cpu"):  
+            with torch.amp.autocast("cuda", enabled=device.type != "cpu"):
                loss = criterion(model(images), labels)

            # Backward