Explorar el Código

Creado con Colaboratory

5.0
Glenn Jocher hace 3 años
padre
commit
b7007d03b4
Se han modificado 1 ficheros con 13 adiciones y 0 borrados
  1. +13
    -0
      tutorial.ipynb

+ 13
- 0
tutorial.ipynb Ver fichero

@@ -1180,6 +1180,19 @@
],
"execution_count": null,
"outputs": []
},
{
"cell_type": "code",
"metadata": {
"id": "BSgFCAcMbk1R"
},
"source": [
"# VOC\n",
"for b, m in zip([64, 48, 32, 16], ['yolov5s', 'yolov5m', 'yolov5l', 'yolov5x']): # zip(batch_size, model)\n",
" !python train.py --batch {b} --weights {m}.pt --data voc.yaml --epochs 50 --cache --img 512 --nosave --hyp hyp.finetune.yaml --project VOC --name {m}"
],
"execution_count": null,
"outputs": []
}
]
}

Cargando…
Cancelar
Guardar