Train simultaneously on two datasets - PyTorch Forums
discuss.pytorch.org › t › train-simultaneously-onFeb 21, 2017 · Hello, I should train using samples from two different datasets, so I initialize two DataLoaders: train_loader_A = torch.utils.data.DataLoader( datasets.ImageFolder(traindir_A), batch_size=args.batch_size, shuffle=True, num_workers=args.workers, pin_memory=True) train_loader_B = torch.utils.data.DataLoader( datasets.ImageFolder(traindir_B), batch_size=args.batch_size, shuffle=True, num_workers ...