From 0a326e7afe3e690c0b4cb64bbd0ce4f0603a7d85 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Tue, 07 Aug 2018 23:45:47 +0000
Subject: [PATCH] XNOR-net on CPU AVX2
---
src/detector.c | 2 ++
1 files changed, 2 insertions(+), 0 deletions(-)
diff --git a/src/detector.c b/src/detector.c
index 244b4c3..a816c74 100644
--- a/src/detector.c
+++ b/src/detector.c
@@ -568,6 +568,7 @@
}
//set_batch_network(&net, 1);
fuse_conv_batchnorm(net);
+ calculate_binary_weights(net);
srand(time(0));
list *plist = get_paths(valid_images);
@@ -1094,6 +1095,7 @@
}
//set_batch_network(&net, 1);
fuse_conv_batchnorm(net);
+ calculate_binary_weights(net);
if (net.layers[net.n - 1].classes != names_size) {
printf(" Error: in the file %s number of names %d that isn't equal to classes=%d in the file %s \n",
name_list, names_size, net.layers[net.n - 1].classes, cfgfile);
--
Gitblit v1.10.0