A bug: num_batches_per_epoch = int(data_size / batch_size) + 1

[lemo2012]

should be:
num_batches_per_epoch = int(math.ceil(float(data_size) / batch_size))