训练中断 IndexError: index 7 is out of bounds for axis 0 with size 7

edvardHua / PoseEstimationForMobile

:dancer: Real-time single person pose estimation for Android and iOS.

Apache License 2.0

1k stars 268 forks source link

训练中断 IndexError: index 7 is out of bounds for axis 0 with size 7 #142

Open gooseable opened 4 years ago

gooseable commented 4 years ago

一次训练到11000次，在train.py中valid_in_image[index,:,:,:]处报错，错误为 IndexError: index 7 is out of bounds for axis 0 with size 7 。改变验证集大小后，训练到12000次中断，报同样的错误。请问这个怎么解决？