我使用Python3.8环境和segmentation_models_pytorch
aka smp
训练了一个分段模型。当我保存它并在我的预测环境中加载(Python3.6和smp
)时,它只在
import torch
model = torch.load(path.join('models', model_name))
但是,它与onnx
包(onnx需要更新的Python)发生冲突。我用Python3.10创建了新的conda环境(另一个使用Python3.11)。现在,torch拒绝装载错误消息ModuleNotFoundError: No module named 'segmentation_models_pytorch.unet'
的模型。
正确的方法是什么?
torch
的torch.load
发布于 2023-06-01 06:03:54
Segmentation_models是一个不同于python的python包,需要单独安装。pip安装-U git+https://github.com/qubvel/segmentation_models.pytorch在python3.10.5及更高版本上为我工作。
发布于 2023-06-01 19:10:18
这是对我有效的解决办法。
将您的SMP降级到您正在进行培训的版本。或者在资源允许的情况下,用新的SMP训练你的模型。这样您就可以避免遗留的Python问题。
因此。这不是火把问题。这是由于SMP模型的改变导致了与dackward兼容性的问题。
https://datascience.stackexchange.com/questions/121878
复制相似问题