我正在尝试查看是否可以将 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 以字节为单位存储(有道理),但我不确定如何解决这个问题:(