diff --git a/Makefile b/Makefile index e0a826eff362d615c44a38d10abda1035bd896be..1aaceefa186813df12b7065ac2eaec51573c56a8 100644 --- a/Makefile +++ b/Makefile @@ -1,6 +1,6 @@ GPU=0 -OPENCV=0 CUDNN=0 +OPENCV=0 DEBUG=0 ARCH= --gpu-architecture=compute_52 --gpu-code=compute_52 diff --git a/cfg/yolo-coco.cfg b/cfg/yolo-coco.cfg index 88438ae973e00b78b03a12401a023d036d389d5a..0c13a3126006ca6c31a7d9af91a76c4436498f48 100644 --- a/cfg/yolo-coco.cfg +++ b/cfg/yolo-coco.cfg @@ -1,6 +1,6 @@ [net] -batch=1 -subdivisions=1 +batch=64 +subdivisions=4 height=448 width=448 channels=3 diff --git a/src/coco.c b/src/coco.c index ef1f12d1635c0b957ddbc4939d71108045dd48f2..af6f7b6318924c86959cfe298b033ccb60d75cf0 100644 --- a/src/coco.c +++ b/src/coco.c @@ -89,7 +89,7 @@ void train_coco(char *cfgfile, char *weightfile) avg_loss = avg_loss*.9 + loss*.1; printf("%d: %f, %f avg, %f rate, %lf seconds, %d images\n", i, loss, avg_loss, get_current_rate(net), sec(clock()-time), i*imgs); - if(i%1000==0){ + if(i%1000==0 || (i < 1000 && i%100 == 0)){ char buff[256]; sprintf(buff, "%s/%s_%d.weights", backup_directory, base, i); save_weights(net, buff);