From a6b2511a566f77a0838dc1dd0d5f3e3c49a8faa0 Mon Sep 17 00:00:00 2001 From: Joseph Redmon Date: Sat, 25 Jun 2016 16:13:54 -0700 Subject: [PATCH] idk --- Makefile | 2 +- cfg/yolo-coco.cfg | 4 ++-- src/coco.c | 2 +- 3 files changed, 4 insertions(+), 4 deletions(-) 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);