idk

Makefile

@@ -1,6 +1,6 @@
 GPU=0
-OPENCV=0
 CUDNN=0
+OPENCV=0
 DEBUG=0
 
 ARCH= --gpu-architecture=compute_52 --gpu-code=compute_52

cfg/yolo-coco.cfg

@@ -1,6 +1,6 @@
 [net]
-batch=1
-subdivisions=1
+batch=64
+subdivisions=4
 height=448
 width=448
 channels=3

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);