使用Pytorch从.ckpt文件加载预训练(CNN)模型

Question

我正在使用Pytorch进行图像分类。我正在寻找在ImageNet以外的数据集上预先训练的CNN模型，我找到了一个指向".ckpt“文件的链接。我还找到了使用Tenserflow加载此文件的教程，但没有使用pytorch。如何从".ckpt“文件中使用Pytorch加载预先训练好的模型？

Answer 1

我同意@jodag的观点，一般来说，PyTorch和Tensorflow是不能互操作的。在某些特殊情况下，您也许能够做到这一点。例如，HuggingFace提供了对转换transformer model from TensorFlow to PyTorch的支持。

有一个related (though closed) question on DataScience StackExchange，其思想是将Tensorflow模型重写到PyTorch中，然后从检查点文件加载权重。请注意，这有时会变得很棘手。