From ae53edc6a42833116b469bdbf288ae9adf01671a Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Thu, 06 Oct 2016 06:35:06 +0000
Subject: [PATCH] fix bug thing
---
src/layer.h | 8 ++++++++
1 files changed, 8 insertions(+), 0 deletions(-)
diff --git a/src/layer.h b/src/layer.h
index 7dbbfb9..ea6862b 100644
--- a/src/layer.h
+++ b/src/layer.h
@@ -4,6 +4,8 @@
#include "activations.h"
#include "stddef.h"
+struct network_state;
+
struct layer;
typedef struct layer layer;
@@ -42,6 +44,12 @@
LAYER_TYPE type;
ACTIVATION activation;
COST_TYPE cost_type;
+ void (*forward) (struct layer, struct network_state);
+ void (*backward) (struct layer, struct network_state);
+ void (*update) (struct layer, int, float, float, float);
+ void (*forward_gpu) (struct layer, struct network_state);
+ void (*backward_gpu) (struct layer, struct network_state);
+ void (*update_gpu) (struct layer, int, float, float, float);
int batch_normalize;
int shortcut;
int batch;
--
Gitblit v1.10.0