添加qat量化支持

2026-01-08 15:12:27 +08:00
parent f4b1f341fc
commit 546a510eb2
8 changed files with 862 additions and 25 deletions
--- a/trainer.py
+++ b/trainer.py
@@ -146,7 +146,6 @@ class YOLO11Trainer:
        iou_thres = 0.7
        iouv = torch.linspace(0.5, 0.95, 10, device=device)
        
-        loss_sum = torch.zeros(3, device=device)
        stats = []
        
        LOGGER.info("\nValidating...")