idk

HaolyShiit · Jun 25, 2016 · a6b2511 · a6b2511
1 parent cf32e7e
commit a6b2511
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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
@@ -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);