From 4373f897f1a5efd1a58746c44430dc33df925415 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Wed, 07 Feb 2018 22:07:19 +0000
Subject: [PATCH] Fixed densenet201_yolo.cfg - burn_in and poly policy, that changes learning rate
---
src/demo.c | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/src/demo.c b/src/demo.c
index c88c1d9..eece00a 100644
--- a/src/demo.c
+++ b/src/demo.c
@@ -125,7 +125,7 @@
demo_classes = classes;
demo_thresh = thresh;
printf("Demo\n");
- net = parse_network_cfg(cfgfile);
+ net = parse_network_cfg_custom(cfgfile, 1);
if(weightfile){
load_weights(&net, weightfile);
}
--
Gitblit v1.10.0