时间戳

首页 > 解决方案 > 收到错误“UnpicklingError：无效的加载键，'v'。” 在 Pytorch 模型中部署在 Streamlit

pytorch - 收到错误“UnpicklingError：无效的加载键，'v'。” 在 Pytorch 模型中部署在 Streamlit

问题描述

我正在尝试在 Streamlit 中部署我的 Pytorch 小麦检测模型。但是，我收到一个错误

UnpicklingError: invalid load key, 'v'.
Traceback:
File "/usr/local/lib/python3.7/site-packages/streamlit/script_runner.py", line 332, in _run_script
    exec(code, module.__dict__)
File "/app/streamlit/streamlit.py", line 84, in <module>
    model.load_state_dict(torch.load(weights_file, map_location=device))
File "/home/appuser/.local/lib/python3.7/site-packages/torch/serialization.py", line 529, in load
    return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)
File "/home/appuser/.local/lib/python3.7/site-packages/torch/serialization.py", line 692, in _legacy_load
    magic_number = pickle_module.load(f, **pickle_load_args)

我的 Github 存储库位于https://github.com/Anubhav1107/streamlit

PL写你想要的。感谢您的帮助！

标签： pytorchpicklestreamlit

解决方案

推荐阅读