From 4373f897f1a5efd1a58746c44430dc33df925415 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Wed, 07 Feb 2018 22:07:19 +0000
Subject: [PATCH] Fixed densenet201_yolo.cfg - burn_in and poly policy, that changes learning rate
---
src/tree.c | 10 ++++++++++
1 files changed, 10 insertions(+), 0 deletions(-)
diff --git a/src/tree.c b/src/tree.c
index cd9fcd1..dfa4178 100644
--- a/src/tree.c
+++ b/src/tree.c
@@ -24,6 +24,16 @@
fprintf(stderr, "Found %d leaves.\n", found);
}
+float get_hierarchy_probability(float *x, tree *hier, int c)
+{
+ float p = 1;
+ while(c >= 0){
+ p = p * x[c];
+ c = hier->parent[c];
+ }
+ return p;
+}
+
void hierarchy_predictions(float *predictions, int n, tree *hier, int only_leaves)
{
int j;
--
Gitblit v1.10.0