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

NO FUCKING SPOILERS DOUG

Browse Source
This commit is contained in:
Joseph Redmon
2018-01-16 14:30:00 -08:00
parent 6e79145309
commit e3931c75cd
23 changed files with 450 additions and 90 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
include/darknet.h
View File

@@ -87,11 +87,12 @@ typedef enum {
REORG,
UPSAMPLE,
LOGXENT,
L2NORM,
BLANK
} LAYER_TYPE;
typedef enum{
SSE, MASKED, L1, SEG, SMOOTH
SSE, MASKED, L1, SEG, SMOOTH,WGAN
} COST_TYPE;
typedef struct{
@@ -162,6 +163,7 @@ struct layer{
float shift;
float ratio;
float learning_rate_scale;
float clip;
int softmax;
int classes;
int coords;
@@ -475,6 +477,7 @@ typedef struct network{
int train;
int index;
float *cost;
float clip;
#ifdef GPU
float *input_gpu;
@@ -604,6 +607,7 @@ void backward_network(network *net);
void update_network(network *net);
float dot_cpu(int N, float *X, int INCX, float *Y, int INCY);
void axpy_cpu(int N, float ALPHA, float *X, int INCX, float *Y, int INCY);
void copy_cpu(int N, float *X, int INCX, float *Y, int INCY);
void scal_cpu(int N, float ALPHA, float *X, int INCX);