Function fn

recipes_source/compiling_optimizer_lr_scheduler.py:65–67

()

@torch.compile(fullgraph=False)
def fn():
    opt.step()
    sched.step()


# Warmup runs to compile the function