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

Question

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

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

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

Answer 1

没有区别。您在Pytorch模型中看到的扩展名是随机的。您可以选择任何东西。
人们通常使用pth来表示 P y T orc H 模型（因此也表示.pth）。但是再一次完全取决于您如何保存模型。