@ptrblck
for epoch_index in range(cfg.epoch):
for batch_idx, (train_batch, val_batch) in enumerate(zip(train_loader, valid_loader)):
data, target = train_batch[0], train_batch[1]
val, target_val = val_batch[0], val_batch[1]
img_batch = variable(data)
label_batch = variable(target).unsqueeze(dim=1)
img_batch_v = variable(val)
label_batch_v = variable(target_val).unsqueeze(dim=1)
For my understanding,  data and target take the data and label form train_loader whereas val and target_val take the value of data and label from valid_loader. Is that correct?
I am facing the problem that train_loader contains 60000 images and valid_loader contains 10000 images.
I think need all the images to take from train_loader and valid_loader.