From 23955b9fa0a29465ad2a2d13c445b49e6d5adef2 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Mon, 08 Feb 2016 19:50:45 +0000
Subject: [PATCH] binary reading weights
---
src/parser.c | 7 +++++--
1 files changed, 5 insertions(+), 2 deletions(-)
diff --git a/src/parser.c b/src/parser.c
index a48f207..8051fd7 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;
}
@@ -810,7 +813,7 @@
{
fprintf(stderr, "Loading weights from %s...", filename);
fflush(stdout);
- FILE *fp = fopen(filename, "r");
+ FILE *fp = fopen(filename, "rb");
if(!fp) file_error(filename);
int major;
--
Gitblit v1.10.0