optimizers.py

import torch.optim

def optimizer(cfg, model):
    result = {"Adam" : torch.optim.Adam(model.parameters(), lr = 1e-2)}
    return result[cfg["Optimizer"]]