Hi-
first - your dataset is a part of your dataloader -
you do not need to mention ‘for batch_A in dataloader(dataset_A)’, but you can: for batchA in dataloader_A
fix this, and show what is your error flag and we can go from there