error，pred_image_on_tensorboard = True

[guker]

PoseEstimationForMobile/training/src/train.py

Line 211 in e31fb85

for index in range(params['batchsize']):

when pred_image_on_tensorboard = True，error as fllow：

IndexError: index 11 is out of bounds for axis 0 with size 11

I have solve it as follow:

PoseEstimationForMobile/training/src/dataset.py

Line 101 in e31fb85

dataset = dataset.batch(batch_size).repeat(epoch)

dataset = dataset.repeat(epoch).batch(batch_size)