From 9bae70b22549b68f5cdeece8b6c3b3de00c22714 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Mon, 16 Apr 2018 23:51:11 +0000
Subject: [PATCH] Accelerated by another 5% using FP16/32 Batch-norm for Tensor Cores.

---
 src/parser.h |    3 +++
 1 files changed, 3 insertions(+), 0 deletions(-)

diff --git a/src/parser.h b/src/parser.h
index b16cc03..b760116 100644
--- a/src/parser.h
+++ b/src/parser.h
@@ -3,8 +3,11 @@
 #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);
 

--
Gitblit v1.10.0