From c53e03348c65462bcba33f6352087dd6b9268e8f Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Wed, 16 Sep 2015 21:12:10 +0000
Subject: [PATCH] yolo working w/ regions
---
src/coco.c | 14 --------------
1 files changed, 0 insertions(+), 14 deletions(-)
diff --git a/src/coco.c b/src/coco.c
index 87f3dca..234f342 100644
--- a/src/coco.c
+++ b/src/coco.c
@@ -111,20 +111,6 @@
avg_loss = avg_loss*.9 + loss*.1;
printf("%d: %f, %f avg, %lf seconds, %d images\n", i, loss, avg_loss, sec(clock()-time), i*imgs);
- if((i-1)*imgs <= N && i*imgs > N){
- fprintf(stderr, "First stage done\n");
- net.learning_rate *= 10;
- char buff[256];
- sprintf(buff, "%s/%s_first_stage.weights", backup_directory, base);
- save_weights(net, buff);
- }
-
- if((i-1)*imgs <= 80*N && i*imgs > N*80){
- fprintf(stderr, "Second stage done.\n");
- char buff[256];
- sprintf(buff, "%s/%s_second_stage.weights", backup_directory, base);
- save_weights(net, buff);
- }
if(i%1000==0){
char buff[256];
sprintf(buff, "%s/%s_%d.weights", backup_directory, base, i);
--
Gitblit v1.10.0