mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

shortcut layers, msr networks

Browse Source
This commit is contained in:
Joseph Redmon
2015-12-14 11:57:10 -08:00
parent 892923514f
commit db0397cfaa
35 changed files with 2635 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/network.h
View File

@ -3,15 +3,15 @@
#define NETWORK_H
#include "image.h"
#include "detection_layer.h"
#include "layer.h"
#include "data.h"
#include "params.h"
typedef enum {
CONSTANT, STEP, EXP, POLY, STEPS, SIG
} learning_rate_policy;
typedef struct {
typedef struct network{
int n;
int batch;
int *seen;
@ -43,6 +43,15 @@ typedef struct {
#endif
} network;
typedef struct network_state {
float *truth;
float *input;
float *delta;
int train;
int index;
network net;
} network_state;
#ifdef GPU
float train_network_datum_gpu(network net, float *x, float *y);
float *network_predict_gpu(network net, float *input);