neuralqx.optimizer.builds module

build_optimizer(cfg, lr)

build_optimizer(cfg: neuralqx.optimizer.types.Adam, lr_spec: neuralqx.optimizer.types.ExponentialDecay | neuralqx.optimizer.types.CosineDecay | neuralqx.optimizer.types.LinearDecay)

build_optimizer(cfg: neuralqx.optimizer.types.SGD, lr: float | neuralqx.optimizer.types.OptaxSchedule)

build_optimizer(cfg: neuralqx.optimizer.types.SGD, lr_spec: neuralqx.optimizer.types.ExponentialDecay | neuralqx.optimizer.types.CosineDecay | neuralqx.optimizer.types.LinearDecay)

build_optimizer(cfg: neuralqx.optimizer.types.Adagrad, lr: float | neuralqx.optimizer.types.OptaxSchedule)

build_optimizer(cfg: neuralqx.optimizer.types.Adagrad, lr_spec: neuralqx.optimizer.types.ExponentialDecay | neuralqx.optimizer.types.CosineDecay | neuralqx.optimizer.types.LinearDecay)

build_optimizer(cfg: neuralqx.optimizer.types.RMSProp, lr: float | neuralqx.optimizer.types.OptaxSchedule)

build_optimizer(cfg: neuralqx.optimizer.types.RMSProp, lr_spec: neuralqx.optimizer.types.ExponentialDecay | neuralqx.optimizer.types.CosineDecay | neuralqx.optimizer.types.LinearDecay)

build_optimizer(cfg: neuralqx.optimizer.types.AdaBelief, lr: float | neuralqx.optimizer.types.OptaxSchedule)

build_optimizer(cfg: neuralqx.optimizer.types.AdaBelief, lr_spec: neuralqx.optimizer.types.ExponentialDecay | neuralqx.optimizer.types.CosineDecay | neuralqx.optimizer.types.LinearDecay)

build_optimizer(cfg: neuralqx.optimizer.types.AMSGrad, lr: float | neuralqx.optimizer.types.OptaxSchedule)

build_optimizer(cfg: neuralqx.optimizer.types.AMSGrad, lr_spec: neuralqx.optimizer.types.ExponentialDecay | neuralqx.optimizer.types.CosineDecay | neuralqx.optimizer.types.LinearDecay)

build_optimizer(cfg: neuralqx.optimizer.types.Yogi, lr: float | neuralqx.optimizer.types.OptaxSchedule)

build_optimizer(cfg: neuralqx.optimizer.types.Yogi, lr_spec: neuralqx.optimizer.types.ExponentialDecay | neuralqx.optimizer.types.CosineDecay | neuralqx.optimizer.types.LinearDecay)