glenn-jocher
commited on
Commit
•
e521909
1
Parent(s):
0cf932b
Created using Colaboratory
Browse files- tutorial.ipynb +2 -2
tutorial.ipynb
CHANGED
@@ -1077,7 +1077,7 @@
|
|
1077 |
},
|
1078 |
"source": [
|
1079 |
"# VOC\n",
|
1080 |
-
"for b, m in zip([64,
|
1081 |
" !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}"
|
1082 |
],
|
1083 |
"execution_count": null,
|
@@ -1099,4 +1099,4 @@
|
|
1099 |
"outputs": []
|
1100 |
}
|
1101 |
]
|
1102 |
-
}
|
|
|
1077 |
},
|
1078 |
"source": [
|
1079 |
"# VOC\n",
|
1080 |
+
"for b, m in zip([64, 64, 32, 16], ['yolov5s', 'yolov5m', 'yolov5l', 'yolov5x']): # zip(batch_size, model)\n",
|
1081 |
" !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}"
|
1082 |
],
|
1083 |
"execution_count": null,
|
|
|
1099 |
"outputs": []
|
1100 |
}
|
1101 |
]
|
1102 |
+
}
|