Ошибка torch.load из-за отсутствия модуля с именем core

Question

Когда я использую torch.load для загрузки какой-либо pth-модели, произошел сбой, и сообщение об ошибке выглядит следующим образом:

Traceback (most recent call last):
  File "torch2onnx.py", line 11, in <module>
    model = torch.load('test.pth')
  File "/home/test/workspace/python-env/torch-venv/lib/python3.6/site-packages/torch/serialization.py", line 426, in load
    return _load(f, map_location, pickle_module, **pickle_load_args)
  File "/home/test/workspace/python-env/torch-venv/lib/python3.6/site-packages/torch/serialization.py", line 613, in _load
    result = unpickler.load()
ModuleNotFoundError: No module named 'core'

Код выглядит следующим образом:

import torch

PTH_PATH='./test.pth'
ONNX_PATH='./test.onnx'

device = torch.device('cpu')
model = torch.load(PTH_PATH, map_location=device)
x = torch.randn(1, 3, 512, 512, requires_grad=True)
torch.onnx.export(model,
                  x,
                  ONNX_PATH,
                  export_params=True,
                  input_names=['input_1'],
                  output_names=['output1'])

Версия факела - 1.30,0