train.optimizer package¶
Submodules¶
train.optimizer.adam module¶
- class train.optimizer.Adam.Adam(**kwargs)¶
Bases:
train.optimizer.optimizer.BaseOptimizer
Adam optimizer
- get_optimizer(weight_params)¶
train.optimizer.mysgd module¶
- class train.optimizer.MySGD.MySGD(**kwargs)¶
Bases:
train.optimizer.optimizer.BaseOptimizer
SGD optimizer
- get_optimizer(weight_params)¶
train.optimizer.optimizer module¶
train.optimizer.rmsprop module¶
- class train.optimizer.RMSprop.RMSprop(**kwargs)¶
Bases:
train.optimizer.optimizer.BaseOptimizer
RMSprop optimizer
- get_optimizer(weight_params)¶
train.optimizer.sgd module¶
- class train.optimizer.SGD.SGD(**kwargs)¶
Bases:
train.optimizer.optimizer.BaseOptimizer
SGD optimizer
- get_optimizer(weight_params)¶