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