# 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