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

Question

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

Answer 1

我同意@jodag的观点，即PyTorch和Tensorflow通常不能互操作。在某些特殊情况下，您可能可以执行此操作。例如，HuggingFace提供了对转换transformer model from TensorFlow to PyTorch的支持。

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