New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

error，pred_image_on_tensorboard = True #133

Open

guker opened this issue Apr 26, 2020 · 0 comments

guker commented Apr 26, 2020 •

edited

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)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment