You mixed up the variable names a bit.
I would change img_batch and label_batch to train_batch and val_batch, respectively.
Could you check, what both loaders are returning, i.e. the shape of both tensor inside the tuple?