def get_optimizer(params, lr=0.001):
    optimizer = torch.optim.AdamW(params, lr=lr, betas=(0.9, 0.999), eps=1e-06, amsgrad=True)
    return optimizer
def get_optimizer(params, lr=0.001):
    optimizer = torch.optim.AdamW(params, lr=lr, betas=(0.9, 0.999), eps=1e-06, amsgrad=True)
    return optimizer