pytorch, 遍历dataloade时,如何获取原始index
pytorch,遍历dataloader的时候
train_loader = DataLoader(train_set, batch_size=batch_size_train, shuffle=True)
因为dataloader已经被shuffle过,
如何知道,他在原始dataset中的
index
torch.utils.data — PyTorch 1.9.1 documentation
pytorch,遍历dataloader的时候
train_loader = DataLoader(train_set, batch_size=batch_size_train, shuffle=True)
因为dataloader已经被shuffle过,
如何知道,他在原始dataset中的
index
torch.utils.data — PyTorch 1.9.1 documentation