Say my forward pass is
def forward(self, x):
x1 = F.relu(F.max_pool2d(self.conv1(x), 2))
x2 = F.relu(F.max_pool2d(self.conv2(x1), 2))
x3 = x2.view(-1, 320)
x5 = self.fc2(x4)
Is there a way i can save the tensor x1.data into a .pt file and use it later as an input for the layers by loading it via dataloader?