idk

Joseph Redmon

2016-06-25 a6b2511a566f77a0838dc1dd0d5f3e3c49a8faa0

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