More than 1 year has passed since last update.

【備忘録】torch.Tensorの保存とロード

Python
PyTorch

Posted at 2023-01-19

import torch

tensor = torch.rand(3)
path = "./mytensor.pt"
torch.save(tensor, path)
loaded_tensor = torch.load(path)

print(tensor == loaded_tensor)
# tensor([True, True, True])

You get articles that match your needs
You can efficiently read back useful information
You can use dark theme

What you can do with signing up