From b36512ea2e7e6d2ec36a3241c1bf751e4e074fe1 Mon Sep 17 00:00:00 2001
From: vinjn <vinjn.z@gmail.com>
Date: Sat, 26 May 2018 13:42:53 +0000
Subject: [PATCH] classifier.c - add the awesome training chart and make sure "top" is not bigger than "classes" in datacfg file.
---
src/route_layer.h | 26 ++++++++------------------
1 files changed, 8 insertions(+), 18 deletions(-)
diff --git a/src/route_layer.h b/src/route_layer.h
index 086ef87..45467d9 100644
--- a/src/route_layer.h
+++ b/src/route_layer.h
@@ -1,28 +1,18 @@
#ifndef ROUTE_LAYER_H
#define ROUTE_LAYER_H
#include "network.h"
+#include "layer.h"
-typedef struct {
- int batch;
- int outputs;
- int n;
- int * input_layers;
- int * input_sizes;
- float * delta;
- float * output;
- #ifdef GPU
- float * delta_gpu;
- float * output_gpu;
- #endif
-} route_layer;
+typedef layer route_layer;
-route_layer *make_route_layer(int batch, int n, int *input_layers, int *input_size);
-void forward_route_layer(const route_layer layer, network net);
-void backward_route_layer(const route_layer layer, network net);
+route_layer make_route_layer(int batch, int n, int *input_layers, int *input_size);
+void forward_route_layer(const route_layer l, network_state state);
+void backward_route_layer(const route_layer l, network_state state);
+void resize_route_layer(route_layer *l, network *net);
#ifdef GPU
-void forward_route_layer_gpu(const route_layer layer, network net);
-void backward_route_layer_gpu(const route_layer layer, network net);
+void forward_route_layer_gpu(const route_layer l, network_state state);
+void backward_route_layer_gpu(const route_layer l, network_state state);
#endif
#endif
--
Gitblit v1.10.0