mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
ADAM
This commit is contained in:
@ -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;
|
||||
|
Reference in New Issue
Block a user