Q:pytorch中的序列化与反序列化
torch.save(obj, f, pickle_module=<module 'pickle' from '/home/v-yuega/miniconda/envs/torch_doc/lib/python3.7/pickle.py'>, pickle_protocol=2, _use_new_zipfile_serialization=True)
torch.load(f, map_location=None, pickle_module=<module 'pickle' from '/home/v-yuega/miniconda/envs/torch_doc/lib/python3.7/pickle.py'>, **pickle_load_args)
Q:模型保存的两种方式是什么?
Q:to函数如何转换数据类型和设备?
x = torch.ones((3, 3)) x = x.to(torch.float64)
x = torch.ones((3, 3)) x = x.to("cuda")
linear = nn.Linear(2, 2) linear.to(torch.double)
gpu1 = torch.device("cuda") linear.to(gpu1)
- 注意:张量不执行inplace,模型执行inplace