Pytorch 中的 .ckpt 和 .pth 文件有什么区别？

Question

我正在关注 GitHub 上使用 Pytorch 的代码。使用以下命令保存模型：

model.save(ARGS.working_dir + '/model_%d.ckpt' % (epoch+1)).  

.pth在 Pytorch 中使用和有什么区别.ckpt？

Answer 1

没有区别。您看到的 Pytorch 模型中的扩展是随机的。你可以选择任何东西。
人们通常用来pth表示P y T orc H模型（因此.pth）。但话又说回来，这完全取决于您想要如何保存模型。