reference/generated/torchrl.trainers.algorithms.configs.utils.LionConfig

Run in Google Colab

Colab

Download Notebook

Notebook

View on GitHub

GitHub

torchrl.trainers.algorithms.configs.utils.LionConfig

class torchrl.trainers.algorithms.configs.utils.LionConfig(lr: float = 0.0001, betas: tuple[float, float] = (0.9, 0.99), weight_decay: float = 0.0, _target_: str = 'torch.optim.Lion', _partial_: bool = True)

Configuration for Lion optimizer.