Skip to content

Instantly share code, notes, and snippets.

@davyneven

davyneven/gold-yolo-train.txt

Created September 28, 2023 11:59
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save davyneven/78269657955bd2d2b7e02ee16a195f29 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save davyneven/78269657955bd2d2b7e02ee16a195f29 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
gold-yolo-train.txt
# train model
IM_SIZE = 640
N_EPOCHS = 50
BATCH_SIZE = 4
!python tools/train.py --img-size $IM_SIZE --epoch $N_EPOCHS --batch $BATCH_SIZE --conf configs/gold_yolo-n.py --data "{project_location}/dataset.yaml" --fuse_ab --device 0
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment