File size: 262 Bytes
cc9c7ee | 1 2 3 4 5 6 7 8 | " Method containing activation functions"
from torch.optim import Adam, AdamW, SGD
from src.utils.mapper import configmapper
configmapper.map("optimizers", "adam")(Adam)
configmapper.map("optimizers", "adam_w")(AdamW)
configmapper.map("optimizers", "sgd")(SGD)
|