From 850f98f5085a7c3425ab91412fbd136b407ab2d0 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 17 Sep 2021 11:39:00 +0200 Subject: [PATCH] Created using Colaboratory --- tutorial.ipynb | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tutorial.ipynb b/tutorial.ipynb index 38e8fd4..6d6a1e7 100644 --- a/tutorial.ipynb +++ b/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,