pytorch_optimizer_config¶
Module pytorch_optimizer_configuration. Specifies a data class for configuring PyTorch optimizers
- class pytorch_optimizer_config.PyTorchOptimizerConfig(optimizer_type: OptimizerType = OptimizerType.ADAM, optimizer_learning_rate: float = 0.01, optimizer_betas: tuple = (0.9, 0.999), optimizer_weight_decay: float = 0, optimizer_amsgrad: bool = False)¶
Configuration class for the optimizer