utils.py 文件源码

def get_optimizer(net, name="Adam"):
    """Get optimizer by name."""
    if name == "Adam":
        return optim.Adam(net.parameters(),
                          lr=params.learning_rate,
                          betas=(params.beta1, params.beta2))