train.learningrate package

Submodules

train.learningrate.CosineAnnealingLR module

class train.learningrate.CosineAnnealingLR.CosineAnnealingLR(**kwargs)

Bases: train.learningrate.learningrate.BaseLearningRate

get_learning_rate()

train.learningrate.ExponentialLR module

class train.learningrate.ExponentialLR.ExponentialLR(**kwargs)

Bases: train.learningrate.learningrate.BaseLearningRate

get_learning_rate()

train.learningrate.MultiStepLR module

class train.learningrate.MultiStepLR.MultiStepLR(**kwargs)

Bases: train.learningrate.learningrate.BaseLearningRate

get_learning_rate()

train.learningrate.StepLR module

class train.learningrate.StepLR.StepLR(**kwargs)

Bases: train.learningrate.learningrate.BaseLearningRate

get_learning_rate()

train.learningrate.learningrate module

class train.learningrate.learningrate.BaseLearningRate(**kwargs)

Bases: object

Abstract class for Learning Rate

get_learning_rate()

Module contents