From 8322a58cf69e79ff1847634ede1a2a1b2e71e6d6 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 14 Jun 2016 18:30:28 +0000
Subject: [PATCH] hate warnings
---
src/classifier.c | 4 ++--
1 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/classifier.c b/src/classifier.c
index 24b28b5..2d0d0e0 100644
--- a/src/classifier.c
+++ b/src/classifier.c
@@ -51,7 +51,7 @@
}
if(clear) *net.seen = 0;
printf("Learning Rate: %g, Momentum: %g, Decay: %g\n", net.learning_rate, net.momentum, net.decay);
- int imgs = net.batch;
+ int imgs = net.batch*net.subdivisions;
list *options = read_data_cfg(datacfg);
@@ -338,10 +338,10 @@
{
int i, j;
network net = parse_network_cfg(filename);
- set_batch_network(&net, 1);
if(weightfile){
load_weights(&net, weightfile);
}
+ set_batch_network(&net, 1);
srand(time(0));
list *options = read_data_cfg(datacfg);
--
Gitblit v1.10.0