Table of Contents ▸ List of Commands ▸ Neural Networks ▸ nn_trainer | ◀ nn_sub | array ▶ |
nn_trainer
Arguments:
- out,in,_optimizer,_"optimizer_params",_scheduler,_"scheduler_params"
Description:
Add a network trainer to the current network, with specified optimizer and scheduler.optimizer can be { sgd | rmsprop | adam | adamax }.
scheduler can be { constant | linear | exponential | adaptive }.
Value of optimizer_params depends on the chosen optimizer:
sgd: (learning_rate,momentum,weight_decay,clip_gradient_min,cli_gradient_max).
rmsprop: (learning_rate,momentum,beta,weight_decay,clip_gradient_min,cli_gradient_max).
adam, adamax: (learning_rate,beta1,beta2,weight_decay,clip_gradient_min,cli_gradient_max).
Default values:
optimizer=adam, optimizer_params=(default), scheduler=constant and scheduler_params=(default).