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

class train.optimizer.optimizer.BaseOptimizer(**kwargs)

Bases: object

Abstract class for Optimizer

get_optimizer(weight_params, params)

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)

Module contents