提交 850f98f5 authored 作者: Glenn Jocher's avatar Glenn Jocher

Created using Colaboratory

上级 27a4736e
......@@ -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,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论