glenn-jocher commited on
Commit
b7007d0
·
1 Parent(s): 9f27902

Creado con Colaboratory

Browse files
Files changed (1) hide show
  1. tutorial.ipynb +13 -0
tutorial.ipynb CHANGED
@@ -1180,6 +1180,19 @@
1180
  ],
1181
  "execution_count": null,
1182
  "outputs": []
 
 
 
 
 
 
 
 
 
 
 
 
 
1183
  }
1184
  ]
1185
  }
 
1180
  ],
1181
  "execution_count": null,
1182
  "outputs": []
1183
+ },
1184
+ {
1185
+ "cell_type": "code",
1186
+ "metadata": {
1187
+ "id": "BSgFCAcMbk1R"
1188
+ },
1189
+ "source": [
1190
+ "# VOC\n",
1191
+ "for b, m in zip([64, 48, 32, 16], ['yolov5s', 'yolov5m', 'yolov5l', 'yolov5x']): # zip(batch_size, model)\n",
1192
+ " !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}"
1193
+ ],
1194
+ "execution_count": null,
1195
+ "outputs": []
1196
  }
1197
  ]
1198
  }