本文整理匯總了Python中torch.serialization方法的典型用法代碼示例。如果您正苦於以下問題：Python torch.serialization方法的具體用法？Python torch.serialization怎麽用？Python torch.serialization使用的例子？那麽, 這裏精選的方法代碼示例或許可以為您提供幫助。您也可以進一步了解該方法所在類torch的用法示例。
在下文中一共展示了torch.serialization方法的3個代碼示例，這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚，您的評價將有助於係統推薦出更棒的Python代碼示例。
示例1: restricted_loads
# 需要導入模塊: import torch [as 別名]
# 或者: from torch import serialization [as 別名]
def restricted_loads(s):
result = RestrictedUnpickler(io.BytesIO(s)).load()
if torch.is_tensor(result) or isinstance(result, torch.nn.Module):
_check_hooks_are_valid(result, "_backward_hooks")
return result
# Adapt torch.load to use RestrictedUnpickler - patched for torch.storage._load_from_bytes
# (Adapted from https://github.com/pytorch/pytorch/blob/master/torch/serialization.py#L602-L773) 示例2: _patch_model_io
# 需要導入模塊: import torch [as 別名]
# 或者: from torch import serialization [as 別名]
def _patch_model_io():
if PatchPyTorchModelIO.__patched:
return
if 'torch' not in sys.modules:
return
PatchPyTorchModelIO.__patched = True
# noinspection PyBroadException
try:
import torch
torch.save = _patched_call(torch.save, PatchPyTorchModelIO._save)
torch.load = _patched_call(torch.load, PatchPyTorchModelIO._load)
# no need to worry about recursive calls, _patched_call takes care of that
if hasattr(torch, 'serialization') and hasattr(torch.serialization, '_save'):
torch.serialization._save = _patched_call(
torch.serialization._save, PatchPyTorchModelIO._save)
if hasattr(torch, 'serialization') and hasattr(torch.serialization, '_load'):
torch.serialization._load = _patched_call(
torch.serialization._load, PatchPyTorchModelIO._load)
if hasattr(torch, 'serialization') and hasattr(torch.serialization, '_legacy_save'):
torch.serialization._legacy_save = _patched_call(
torch.serialization._legacy_save, PatchPyTorchModelIO._save)
if hasattr(torch, 'serialization') and hasattr(torch.serialization, '_legacy_load'):
torch.serialization._legacy_load = _patched_call(
torch.serialization._legacy_load, PatchPyTorchModelIO._load)
except ImportError:
pass
except Exception:
pass # print('Failed patching pytorch') 示例3: build_model
# 需要導入模塊: import torch [as 別名]
# 或者: from torch import serialization [as 別名]
def build_model(self, args):
model = super().build_model(args)
if args.pretrained is not None: # load pretrained model:
if not os.path.exists(args.pretrained):
raise ValueError('Could not load pretrained weights \
- from {}'.format(args.pretrained))
from torch.serialization import default_restore_location
saved_state = torch.load(
args.pretrained,
map_location=lambda s, l: default_restore_location(s, 'cpu')
)
self.adapt_state(saved_state['model'], model)
return model