update readme

1c4460ad · Bubbliiiing · b995821d · 1c4460ad
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

summary.py summary.py +1 -1

未找到文件。
--- a/summary.py
+++ b/summary.py
@@ -11,7 +11,7 @@ if __name__ == "__main__":
    input_shape     = [640, 640]
    anchors_mask    = [[6, 7, 8], [3, 4, 5], [0, 1, 2]]
    num_classes     = 80
-    phi             = 'yolov7'
+    phi             = 'l'
    
    device  = torch.device("cuda" if torch.cuda.is_available() else "cpu")
    m       = YoloBody(anchors_mask, num_classes, phi, False).to(device)