def linear_warmup()

in train_helpers.py [0:0]

• 4 lines of code
• 2 McCabe index (conditional complexity)

def linear_warmup(warmup_iters):
    def f(iteration):
        return 1.0 if iteration > warmup_iters else iteration / warmup_iters
    return f