Public Attributes | |
unsigned int | n_epochs |
unsigned int | batch_size |
double | learning_rate |
unsigned int magmadnn::model::nn_params_t::batch_size |
batch_size the size of the batch
double magmadnn::model::nn_params_t::learning_rate |
initial learning rate
unsigned int magmadnn::model::nn_params_t::n_epochs |
n_epochs number of epochs to train for