" 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)