Module: tfm.optimization.opt_cfg

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.