Function get_lr

diff2flow/lr_schedulers.py:121–123

(optimizer)

    import matplotlib.pyplot as plt

    def get_lr(optimizer):
        for param_group in optimizer.param_groups:
            return param_group['lr']

    net = nn.Linear(10, 10)
    opt = torch.optim.Adam(net.parameters(), lr=3e-5)