From 2afa376bb37b379f27954f74859fbfa63402ea46 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Fri, 14 Aug 2015 18:45:11 +0000
Subject: [PATCH] single shot yolo training, separate file
---
src/parser.c | 3 +++
1 files changed, 3 insertions(+), 0 deletions(-)
diff --git a/src/parser.c b/src/parser.c
index 5591dc3..b373c01 100644
--- a/src/parser.c
+++ b/src/parser.c
@@ -236,6 +236,9 @@
{
float probability = option_find_float(options, "probability", .5);
dropout_layer layer = make_dropout_layer(params.batch, params.inputs, probability);
+ layer.out_w = params.w;
+ layer.out_h = params.h;
+ layer.out_c = params.c;
return layer;
}
--
Gitblit v1.10.0