Parcourir la source

Update tutorial.ipynb (#7212)

modifyDataloader
Glenn Jocher GitHub il y a 2 ans
Parent
révision
9c1e91aea2
Aucune clé connue n'a été trouvée dans la base pour cette signature ID de la clé GPG: 4AEE18F83AFDEB23
1 fichiers modifiés avec 1 ajouts et 1 suppressions
  1. +1
    -1
      tutorial.ipynb

+ 1
- 1
tutorial.ipynb Voir le fichier

@@ -1077,7 +1077,7 @@
},
"source": [
"# VOC\n",
"for b, m in zip([64, 64, 32, 16], ['yolov5s', 'yolov5m', 'yolov5l', 'yolov5x']): # zip(batch_size, model)\n",
"for b, m in zip([64, 64, 64, 32, 16], ['yolov5n', '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.VOC.yaml --project VOC --name {m}"
],
"execution_count": null,

Chargement…
Annuler
Enregistrer