File size: 503 Bytes
cc9c7ee |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
from torch.optim.lr_scheduler import (
StepLR,
CosineAnnealingLR,
ReduceLROnPlateau,
CyclicLR,
CosineAnnealingWarmRestarts,
)
from src.utils.mapper import configmapper
configmapper.map("schedulers", "step")(StepLR)
configmapper.map("schedulers", "cosineanneal")(CosineAnnealingLR)
configmapper.map("schedulers", "reduceplateau")(ReduceLROnPlateau)
configmapper.map("schedulers", "cyclic")(CyclicLR)
configmapper.map("schedulers", "cosineannealrestart")(CosineAnnealingWarmRestarts)
|