ride.optimizers
¶
Modules adding optimizers
Module Contents¶
Classes¶
Abstract base-class for Optimizer mixins |
|
Abstract base-class for Optimizer mixins |
|
Abstract base-class for Optimizer mixins |
|
Abstract base-class for Optimizer mixins |
|
Abstract base-class for Optimizer mixins |
|
Abstract base-class for Optimizer mixins |
|
Abstract base-class for Optimizer mixins |
|
Abstract base-class for Optimizer mixins |
|
Abstract base-class for Optimizer mixins |
|
Abstract base-class for Optimizer mixins |
Functions¶
|
|
|
- ride.optimizers.discounted_steps_per_epoch(base_steps: int, num_gpus: int, accumulate_grad_batches: int)[source]¶
- class ride.optimizers.SgdOptimizer(hparams: pytorch_lightning.utilities.parsing.AttributeDict, *args, **kwargs)[source]¶
Bases:
ride.core.OptimizerMixin
Abstract base-class for Optimizer mixins
- static configs() ride.core.Configs [source]¶
- class ride.optimizers.AdamWOptimizer(hparams: pytorch_lightning.utilities.parsing.AttributeDict, *args, **kwargs)[source]¶
Bases:
ride.core.OptimizerMixin
Abstract base-class for Optimizer mixins
- static configs() ride.core.Configs [source]¶
- class ride.optimizers.SgdReduceLrOnPlateauOptimizer(hparams: pytorch_lightning.utilities.parsing.AttributeDict, *args, **kwargs)[source]¶
Bases:
ride.core.OptimizerMixin
Abstract base-class for Optimizer mixins
- static configs() ride.core.Configs [source]¶
- class ride.optimizers.AdamWReduceLrOnPlateauOptimizer(hparams: pytorch_lightning.utilities.parsing.AttributeDict, *args, **kwargs)[source]¶
Bases:
ride.core.OptimizerMixin
Abstract base-class for Optimizer mixins
- static configs() ride.core.Configs [source]¶
- class ride.optimizers.SgdCyclicLrOptimizer(hparams: pytorch_lightning.utilities.parsing.AttributeDict, *args, **kwargs)[source]¶
Bases:
ride.core.OptimizerMixin
Abstract base-class for Optimizer mixins
- static configs() ride.core.Configs [source]¶
- class ride.optimizers.AdamWCyclicLrOptimizer(hparams: pytorch_lightning.utilities.parsing.AttributeDict, *args, **kwargs)[source]¶
Bases:
ride.core.OptimizerMixin
Abstract base-class for Optimizer mixins
- static configs() ride.core.Configs [source]¶
- class ride.optimizers.SgdOneCycleOptimizer(hparams: pytorch_lightning.utilities.parsing.AttributeDict, *args, **kwargs)[source]¶
Bases:
ride.core.OptimizerMixin
Abstract base-class for Optimizer mixins
- static configs() ride.core.Configs [source]¶
- class ride.optimizers.AdamWOneCycleOptimizer(hparams: pytorch_lightning.utilities.parsing.AttributeDict, *args, **kwargs)[source]¶
Bases:
ride.core.OptimizerMixin
Abstract base-class for Optimizer mixins
- static configs() ride.core.Configs [source]¶
- class ride.optimizers.SgdMultiStepLR(hparams: pytorch_lightning.utilities.parsing.AttributeDict, *args, **kwargs)[source]¶
Bases:
ride.core.OptimizerMixin
Abstract base-class for Optimizer mixins
- static configs() ride.core.Configs [source]¶
- class ride.optimizers.AdamWMultiStepLR(hparams: pytorch_lightning.utilities.parsing.AttributeDict, *args, **kwargs)[source]¶
Bases:
ride.core.OptimizerMixin
Abstract base-class for Optimizer mixins
- static configs() ride.core.Configs [source]¶
- ride.optimizers.discriminative_lr_and_params(model: torch.nn.Module, lr: float, discriminative_lr_fraction: float)[source]¶