Apache SINGA
A distributed deep learning platform .
|
ParamEntry is used for aggregating gradients of Params shared by workers from the same group. More...
#include <param.h>
Public Member Functions | |
ParamEntry (int total, Param *p) | |
void | AddParam (bool local, Param *p) |
Associate the counter to a Param object. More... | |
Public Attributes | |
int | next_version = -1 |
int | num_update = 0 |
int | num_local = 0 |
local workers using the shared parameter | |
int | num_total = 0 |
total workers using the shared parameterMore... | |
std::vector< Param * > | shares |
ParamEntry is used for aggregating gradients of Params shared by workers from the same group.
For each worker group, every unique Param object has a ParamEntry object. Param objects sharing the same values are associated with the same ParamEntry.
void singa::ParamEntry::AddParam | ( | bool | local, |
Param * | p | ||
) |
Associate the counter to a Param object.
p | |
local | 1 if it is used by workers in this procs, 0 otherwise |
int singa::ParamEntry::num_total = 0 |
Shares are deleted by neuralnet's destructor