Created using Colaboratory

tutorial.ipynb

@@ -957,7 +957,6 @@
         "# Unit tests\n",
         "%%shell\n",
         "export PYTHONPATH=\"$PWD\"  # to run *.py. files in subdirectories\n",
-        "\n",
         "rm -rf runs  # remove runs/\n",
         "for m in yolov5s; do  # models\n",
         "  python train.py --weights $m.pt --epochs 3 --img 320 --device 0  # train pretrained\n",
@@ -968,9 +967,10 @@
         "    python val.py --weights $m.pt --device $d # val official\n",
         "    python val.py --weights runs/train/exp/weights/best.pt --device $d # val custom\n",
         "  done\n",
-        "  python hubconf.py  # hub\n",
-        "  python models/yolo.py --cfg $m.yaml  # inspect\n",
-        "  python export.py --weights $m.pt --img 640 --batch 1  # export\n",
+        "python hubconf.py  # hub\n",
+        "python models/yolo.py --cfg $m.yaml  # build PyTorch model\n",
+        "python models/tf.py --weights $m.pt  # build TensorFlow model\n",
+        "python export.py --img 128 --batch 1 --weights $m.pt --include torchscript onnx  # export\n",
         "done"
       ],
       "execution_count": null,