gpt4 book ai didi

pytorch - Pytorch 中的 .ckpt 和 .pth 文件有什么区别?

转载 作者:行者123 更新时间:2023-12-04 03:50:39 25 4
gpt4 key购买 nike

我正在关注 GitHub 中使用 Pytorch 的代码。
使用以下方法保存模型:

model.save(ARGS.working_dir + '/model_%d.ckpt' % (epoch+1)).  
使用 .pth 有什么区别和 .ckpt在 Pytorch 中?

最佳答案

没有区别。您看到的 Pytorch 模型中的扩展是随机的。你可以选择任何东西。
人们通常使用pth表示 电话 y 电话 兽人 H 模型(因此 .pth )。但又完全取决于您想如何保存模型。

关于pytorch - Pytorch 中的 .ckpt 和 .pth 文件有什么区别?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/64456843/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com