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