From 6aa140b3c6e9917936f73bd538fce4fe20954233 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Sun, 15 Apr 2018 13:57:05 +0000
Subject: [PATCH] Disable burn_in for FP16 Tensor Cores
---
src/region_layer.c | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/src/region_layer.c b/src/region_layer.c
index d99636d..4aff75a 100644
--- a/src/region_layer.c
+++ b/src/region_layer.c
@@ -544,7 +544,7 @@
int class_index = entry_index(l, 0, n*l.w*l.h + i, l.coords + !l.background);
if (l.softmax_tree) {
- hierarchy_predictions(predictions + class_index, l.classes, l.softmax_tree, 0, l.w*l.h);
+ hierarchy_predictions(predictions + class_index, l.classes, l.softmax_tree, 0);// , l.w*l.h);
if (map) {
for (j = 0; j < 200; ++j) {
int class_index = entry_index(l, 0, n*l.w*l.h + i, l.coords + 1 + map[j]);
--
Gitblit v1.10.0