From 70433a2701e9db85ba38ce1a2052179e9133cdfb Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 30 Mar 2022 12:53:49 +0200 Subject: [PATCH] Update tutorial.ipynb (#7212) --- tutorial.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tutorial.ipynb b/tutorial.ipynb index 1479a164cd8e..0379fb1a3c57 100644 --- a/tutorial.ipynb +++ b/tutorial.ipynb @@ -1077,7 +1077,7 @@ }, "source": [ "# VOC\n", - "for b, m in zip([64, 64, 32, 16], ['yolov5s', 'yolov5m', 'yolov5l', 'yolov5x']): # zip(batch_size, model)\n", + "for b, m in zip([64, 64, 64, 32, 16], ['yolov5n', 'yolov5s', 'yolov5m', 'yolov5l', 'yolov5x']): # zip(batch_size, model)\n", " !python train.py --batch {b} --weights {m}.pt --data VOC.yaml --epochs 50 --cache --img 512 --nosave --hyp hyp.VOC.yaml --project VOC --name {m}" ], "execution_count": null,