From 01f20abe847768a8fb3e27739a284b541eb08ff6 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Fri, 05 Feb 2016 20:56:18 +0000
Subject: [PATCH] generate
---
src/parser.c | 5 ++++-
1 files changed, 4 insertions(+), 1 deletions(-)
diff --git a/src/parser.c b/src/parser.c
index a48f207..68a9f04 100644
--- a/src/parser.c
+++ b/src/parser.c
@@ -176,8 +176,11 @@
char *activation_s = option_find_str(options, "activation", "logistic");
ACTIVATION activation = get_activation(activation_s);
int batch_normalize = option_find_int_quiet(options, "batch_normalize", 0);
+ int logistic = option_find_int_quiet(options, "logistic", 0);
- layer l = make_rnn_layer(params.batch, params.inputs, hidden, output, params.time_steps, activation, batch_normalize);
+ layer l = make_rnn_layer(params.batch, params.inputs, hidden, output, params.time_steps, activation, batch_normalize, logistic);
+
+ l.shortcut = option_find_int_quiet(options, "shortcut", 0);
return l;
}
--
Gitblit v1.10.0