如何将 Spacy Model .pkl 文件转换为 .pt/.pth pytorch 支持的格式

Question

我有 spacy 模型，用于 .pkl 格式的推理。 .pkl 文件的数据类型是 。我想让推理脚本在 GPU 上运行。我尝试使用 spacy gpu、numba 等不同的方法。

import spacy  
spacy.prefer_gpu() # or spacy.require_gpu()
nlp = spacy.load("content/path")

它不起作用，我认为通过将 .pkl 转换为 .pt 将通过将 pt 文件加载到“cuda”设备来工作。请提出处理这种情况的方法。

Answer 1

将 .pkl 文件转换为 .pt 或 .pth 文件的一种迂回方法是展平 .pkl 文件，然后初始化具有相同形状的新模型。如果您随后重新加载展平权重并将其保存到 .pth，它可能会起作用。