Created using Colaboratory

2022-08-16 22:22:52 +02:00 · 2022-08-16 22:22:52 +02:00 · d1dfcab604
parent 464c2c6713
commit d1dfcab604
1 changed files with 1 additions and 1 deletions
--- 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}"
      ],