for it, (images_a, images_b) in enumerate(zip(train_loader_a,train_loader_b)):
The error happens in above line. I suppose it is caused by the dataloader, and find that other people also meet this problem but no solution.