The dataloader I have is derived like this:
from torch.utils.data import DataLoader,TensorDataset
dataset = TensorDataset(tensor0, tensor1)
dataloader = DataLoader(
dataset,
sampler = RandomSampler(dataset)
batch_size = batch_size
)
Then I realized I had to add tensor2
when creating dataset: It should have been
dataset = TensorDataset(tensor0, tensor1, tensor2)
But I didn’t save tensor0
, tensor1
, or dataset
. Creating these all over again is very costly.
Can I make the desired modification to dataloader
if I only have dataloader
?