def gen_epoch_data(num_epochs, batch_size, seq_length): for i in range(num_epochs): yield reader.ptb_iterator(data, batch_size, seq_length)