def METHOD_NAME(session, iters):
    for i, batch in enumerate(train_loader):
        x = batch[0].detach().cpu().numpy()
        in_tensor = {'input': x}
        session.run(None, in_tensor)
        if i >= iters:
            break