Parameter | Type | Description | Example |
---|---|---|---|
object | The information about the learning rate scheduler. | ||
Gamma | float | The learning rate decay. This parameter takes effect only when LRScheduler is set to StepLR. | 0.97 |
StepSize | long | The number of epochs the learning rate is changed after. This parameter takes effect only when LRScheduler is set to StepLR. | 1 |
LRScheduler | string | The learning rate scheduler. | StepLR |