def get_optim_scheduler()

in 01-byoc/code/workflow.py [0:0]

• 6 lines of code
• 3 McCabe index (conditional complexity)

def get_optim_scheduler(params, model):
    # optimizer
    if params.optimizer == "adam":
        optimizer = Adam(model.parameters(), lr=params.lr)
    # scheduler
    if params.scheduler == "steplr":
        scheduler = StepLR(optimizer, step_size=int(params.epochs*0.8), gamma=0.1)
    return optimizer, scheduler