From 0f7f2899b65343e56b0a1188f703d459d824d398 Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Mon, 16 Nov 2015 03:51:26 +0000 Subject: [PATCH] Fix for cuda 7.5 --- src/parser.h | 2 ++ 1 files changed, 2 insertions(+), 0 deletions(-) diff --git a/src/parser.h b/src/parser.h index b16cc03..6cff4fb 100644 --- a/src/parser.h +++ b/src/parser.h @@ -5,6 +5,8 @@ network parse_network_cfg(char *filename); void save_network(network net, char *filename); void save_weights(network net, char *filename); +void save_weights_upto(network net, char *filename, int cutoff); +void save_weights_double(network net, char *filename); void load_weights(network *net, char *filename); void load_weights_upto(network *net, char *filename, int cutoff); -- Gitblit v1.10.0