From 989ab8c38a02fa7ea9c25108151736c62e81c972 Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Fri, 24 Apr 2015 17:27:50 +0000 Subject: [PATCH] IOU loss function --- src/parser.h | 3 +++ 1 files changed, 3 insertions(+), 0 deletions(-) diff --git a/src/parser.h b/src/parser.h index 891e658..b16cc03 100644 --- a/src/parser.h +++ b/src/parser.h @@ -4,5 +4,8 @@ network parse_network_cfg(char *filename); void save_network(network net, char *filename); +void save_weights(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