Module audiocraft.optim

Optimization stuff. In particular, optimizers (DAdaptAdam), schedulers and Exponential Moving Average.

Sub-modules

audiocraft.optim.cosine_lr_scheduler

audiocraft.optim.dadam

audiocraft.optim.ema

audiocraft.optim.fsdp

Wrapper around FSDP for more convenient use in the training loops.

audiocraft.optim.inverse_sqrt_lr_scheduler

audiocraft.optim.linear_warmup_lr_scheduler

audiocraft.optim.polynomial_decay_lr_scheduler