From cdd1cb0e8c4fda3671714bb5ad6ba1825cff16d1 Mon Sep 17 00:00:00 2001 From: Puneet Kohli <punkohl@gmail.com> Date: Sun, 06 May 2018 21:45:26 +0000 Subject: [PATCH] Change matplotlib backend to 'agg' when 'show_plot' is false. This allows the script to be run from command line without any errors (for ex, on a remote machine) --- src/parser.h | 6 ++++++ 1 files changed, 6 insertions(+), 0 deletions(-) diff --git a/src/parser.h b/src/parser.h index 891e658..b760116 100644 --- a/src/parser.h +++ b/src/parser.h @@ -3,6 +3,12 @@ #include "network.h" network parse_network_cfg(char *filename); +network parse_network_cfg_custom(char *filename, int batch); 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); #endif -- Gitblit v1.10.0