def METHOD_NAME(batch_idx, loss, output, target):
    # print train info
    global prev
    pred = np.argmax(output, axis=1)
    acc = np.mean(target==pred)
    loss_list.append(loss[0])
    acc_list.append(acc)
    print('Train Epoch: {} [{}/{} ({:.0f}%)]\tLoss: {:.6f}\tAcc: {:.6f} \tTime:{:.3f}'
        .format(0, batch_idx, 300,1. * batch_idx / 6.0, loss[0], acc, time.time()-prev))
    # prev = time.time()