From d8adaf8ea6a31a380f6bf1fe65e88b661d3bb51e Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Fri, 21 Oct 2016 20:16:43 +0000
Subject: [PATCH] tree stuff

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

diff --git a/src/layer.h b/src/layer.h
index ea6862b..341e58a 100644
--- a/src/layer.h
+++ b/src/layer.h
@@ -3,6 +3,7 @@
 
 #include "activations.h"
 #include "stddef.h"
+#include "tree.h"
 
 struct network_state;
 
@@ -93,6 +94,8 @@
     int reorg;
     int log;
 
+    tree *softmax_tree;
+
     float alpha;
     float beta;
     float kappa;

--
Gitblit v1.10.0