art, cudnn

This commit is contained in:
Joseph Redmon
2016-05-13 11:59:43 -07:00
parent 054e2b1954
commit 13209df7bb
11 changed files with 286 additions and 30 deletions

View File

@ -11,6 +11,7 @@ typedef enum {
} learning_rate_policy;
typedef struct network{
float *workspace;
int n;
int batch;
int *seen;
@ -49,6 +50,7 @@ typedef struct network_state {
float *truth;
float *input;
float *delta;
float *workspace;
int train;
int index;
network net;