Shortcuts

torchrl.trainers.algorithms.configs.utils.NAdamConfig

class torchrl.trainers.algorithms.configs.utils.NAdamConfig(lr: float = 0.002, betas: tuple[float, float] = (0.9, 0.999), eps: float = 1e-08, weight_decay: float = 0.0, momentum_decay: float = 0.004, foreach: bool | None = None, _target_: str = 'torch.optim.NAdam', _partial_: bool = True)[source]

Configuration for NAdam optimizer.

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources