Well I’m sure at that time, the batch size is 28, you can print both data and target shape to verify
PS, quick fix is to simply add “if data_batch_size < 64 or target_batch_size < 64: continue”
or replace  range(64) with range(len(target.data)). Boom done, you don’t need to care what the size is