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/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