There is no more information unfortunately, and no info about which line is causing the error. However, when I debug and try to get the first batch of (data, label) in the for loop, it causes this error.