train.py 文件源码

def get_current_learning_rate(opt):
    if isinstance(opt, optimizers.NesterovAG):
        return opt.lr
    if isinstance(opt, optimizers.Adam):
        return opt.alpha
    if isinstance(opt, optimizers.SGD):
        return opt.lr
    raise NotImplementationError()