From c123f320fcc2212f1d3b59a4d3d2ae6be71cf1e6 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Sun, 04 Mar 2018 20:41:23 +0000
Subject: [PATCH] Show avg-loss chart during Training, if compiled with OpenCV. Use -dont_show to disable.

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

diff --git a/src/layer.h b/src/layer.h
index 285abe3..9bf6057 100644
--- a/src/layer.h
+++ b/src/layer.h
@@ -34,6 +34,7 @@
     XNOR,
     REGION,
     REORG,
+	REORG_OLD,
     BLANK
 } LAYER_TYPE;
 
@@ -63,6 +64,7 @@
     int out_h, out_w, out_c;
     int n;
     int max_boxes;
+	int small_object;
     int groups;
     int size;
     int side;
@@ -241,6 +243,9 @@
     float * weights_gpu;
     float * weight_updates_gpu;
 
+	float * weights_gpu16;
+	float * weight_updates_gpu16;
+
     float * biases_gpu;
     float * bias_updates_gpu;
 

--
Gitblit v1.10.0