no it is not dataloader is the entire dataset and train_loader is just the train set. My focus is on train_loader
When I do that I get a tensor of 600 values for mean and same for std. I only should have 3 values for each
Screenshot_2020-10-14_23-30-552534×1496 380 KB