memcache pytorch 预训练模型

Question

我正在尝试查看是否可以将 pytorch 模型加载到 memcache 并在稍后的会话中使用它。所以，我所拥有的是来自这里： https://pytorch.org/hub/pytorch_vision_mobilenet_v2/

我像这样加载这个模型到内存缓存：

import torch
_model = torch.hub.load('pytorch/vision:v0.9.0', 'mobilenet_v2', pretrained=True)
_model.eval()

client.set('model', _model)
model=client.get('model')

它似乎获取模型，但是当我运行它时，我得到：

out = model(input_batch)
TypeError: 'bytes' object is not callable

我认为 memcache 以字节为单位存储（有道理），但我不确定如何解决这个问题:(