gigl.src.common.utils.scheduler

Classes

CosineDecayScheduler

A cosine decay scheduler that is used instead of torch.optim.lr_scheduler.CosineAnnealingLR since we need a scheduler specifically for EMA weight averaging (momentum).