为什么我们需要使用 .net 扩展名保存 pytorch 模型?

疼痛

我是 Pytorch 的新学习者,我正在研究 Character_Level_LSTM_Exercise。

为什么他们在模型名称中使用 .net 扩展名保存模型?

我正在寻找解释,但没有得到任何好的解释。

# change the name, for saving multiple files
model_name = 'rnn_x_epoch.net'

checkpoint = {'n_hidden': net.n_hidden,
              'n_layers': net.n_layers,
              'state_dict': net.state_dict(),
              'tokens': net.chars}

with open(model_name, 'wb') as f:
    torch.save(checkpoint, f)
莫贝克

您可以使用任何您喜欢的扩展名!只要确保一致。

文档建议使用 .pt 扩展名。

https://pytorch.org/docs/stable/generated/torch.save.html

有关更多解释和更多扩展选项,请参阅Soumith Chintala 的评论。

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章

为什么我们需要在PyTorch中调用zero_grad()?

使用JFileChooser设置默认保存扩展名

Vim为什么保存带有〜扩展名的文件?

Pytorch保存和重新加载模型

如何使用PyTorch训练原始的U-Net模型?

如何从我的网页保存Arduino代码“ .ino扩展名”?

为什么我们不再需要在更高版本的ASP.NET Core中手动验证模型?

保存PyTorch模型而无法访问模型类代码

如何在PyTorch中保存模型架构?

为什么pytorch模型无法识别我定义的张量?

为什么PyTorch模型在模型中需要多个图像大小?

无法使用经过训练的NSGA-Net PyTorch模型生成对抗性示例

使用AngularJs + Web API,为什么我们需要ASP.NET MVC和IIS?

我应该使用.sh扩展名保存脚本吗?

为什么我们需要对authorize.net库使用curl?

iOS联系人备份我们需要保存的文件扩展名

当我们总是从uvm_sequence_item扩展时,为什么要使用uvm_transaction类呢?

使用.NET sosex扩展名的注册

为什么我们使用扩展?

使用 Pillow 保存 tkinter 画布会保存无扩展名的文件,但不会指定扩展名

需要使用 javascript(.sql 文件扩展名)在文件中保存 textArea 值

为什么我们使用损失来更新我们的模型,但使用指标来选择我们需要的模型?

我们为什么要使用扩展?

如何在 PyTorch 中保存模型?

PyTorch:保存权重和模型定义

当我们可以通过共享扩展 SQL DB 时,为什么要使用 NoSQL DB?

pytorch 保存和加載模型

即使我们使用expressjs更改扩展名,如何获取原始扩展名

为什么我们需要从 PyTorch 中的 nn.Module 继承?