generate_batches.py issue

Hello Andrew, I have a problem with generating batches, sometimes the error jumps out like the image that I show, but sometimes it won't, the batches that I successfully created are workable on Hierarchical-CADNet, Could you help me to solve this error. Thanking you