diff --git a/Makefile b/Makefile index e0a826ef..1aaceefa 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 88438ae9..0c13a312 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 ef1f12d1..af6f7b63 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);