Saving and loading tensors as training data

1

I have created my training data for a CNN and would like to save it for future training.
How would I go about saving my training data and then loading it again?

My training data looks like:

X.shape
Out[346]: torch.Size([30, 1, 362, 362])

y.shape
Out[347]: torch.Size([30, 131044, 3])
2

torch.save and torch.load should work. :slight_smile:

3

That works,

cheers :slight_smile: