Browse Source

Created using Colaboratory

modifyDataloader
Glenn Jocher 2 years ago
parent
commit
e5219099cd
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      tutorial.ipynb

+ 2
- 2
tutorial.ipynb View File

@@ -1077,7 +1077,7 @@
},
"source": [
"# VOC\n",
"for b, m in zip([64, 48, 32, 16], ['yolov5s', 'yolov5m', 'yolov5l', 'yolov5x']): # zip(batch_size, model)\n",
"for b, m in zip([64, 64, 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,
@@ -1099,4 +1099,4 @@
"outputs": []
}
]
}
}

Loading…
Cancel
Save