def METHOD_NAME(data, model, loss_func, optimizer, device, g_data):
    model.train()
    out = model(g_data)
    loss = get_loss("train", data, out, loss_func, device, get_auc=False)
    optimizer.zero_grad()
    loss.backward()
    optimizer.step()
    optimizer.zero_grad()
    return g_data