This commit is contained in:
Joseph Redmon
2016-10-26 08:35:44 -07:00
parent 91f95c715b
commit 352ae7e65b
11 changed files with 114 additions and 26 deletions

View File

@ -94,6 +94,14 @@ struct layer{
int reorg;
int log;
int adam;
float B1;
float B2;
float eps;
float *m_gpu;
float *v_gpu;
int t;
tree *softmax_tree;
float alpha;