Function get_lr

src/train_new.py:156–158

(optimizer)

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


# define the training function.