From a6b2511a566f77a0838dc1dd0d5f3e3c49a8faa0 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Sat, 25 Jun 2016 23:13:54 +0000
Subject: [PATCH] idk
---
src/coco.c | 2 +-
Makefile | 2 +-
cfg/yolo-coco.cfg | 4 ++--
3 files changed, 4 insertions(+), 4 deletions(-)
diff --git a/Makefile b/Makefile
index e0a826e..1aaceef 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 88438ae..0c13a31 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 ef1f12d..af6f7b6 100644
--- a/src/coco.c
+++ b/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);
--
Gitblit v1.10.0