Created using Colaboratory

ultralytics · Aug 16, 2022 · d1dfcab · d1dfcab
1 parent 464c2c6
commit d1dfcab
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tutorial.ipynb b/tutorial.ipynb
@@ -1113,7 +1113,7 @@
       "cell_type": "code",
       "source": [
         "# Classification\n",
-        "for m in *(f'yolov5{x}.pt' for x in 'nsmlx'), 'resnet50.pt', 'efficientnet_b0.pt':\n",
+        "for m in [*(f'yolov5{x}.pt' for x in 'nsmlx'), 'resnet50.pt', 'efficientnet_b0.pt']:\n",
         "  for d in 'mnist', 'fashion-mnist', 'cifar10', 'cifar100', 'imagenette160', 'imagenette320', 'imagenette', 'imagewoof160', 'imagewoof320', 'imagewoof':\n",
         "    !python classify/train.py --model {m} --data {d} --epochs 10 --project YOLOv5-cls --name {m}-{d}"
       ],