View source on GitHub
|
Dataclasses for optimizer configs.
Classes
class AdafactorConfig: Configuration for Adafactor optimizer.
class AdagradConfig: Configuration for Adagrad optimizer.
class AdamConfig: Configuration for Adam optimizer.
class AdamExperimentalConfig: Configuration for experimental Adam optimizer.
class AdamWeightDecayConfig: Configuration for Adam optimizer with weight decay.
class AdamWeightDecayExperimentalConfig: Configuration for Adam optimizer with weight decay.
class BaseOptimizerConfig: Base optimizer config.
class EMAConfig: Exponential moving average optimizer config.
class LAMBConfig: Configuration for LAMB optimizer.
class LARSConfig: Layer-wise adaptive rate scaling config.
class RMSPropConfig: Configuration for RMSProp optimizer.
class SGDConfig: Configuration for SGD optimizer.
class SGDExperimentalConfig: Configuration for SGD optimizer.
class SLIDEConfig: Configuration for SLIDE optimizer.