如何归档训练有素的 PyTorch 模型?

数据挖掘 数据科学模型 火炬 模型选择
2022-03-01 13:47:03

目前,我正在研究基于LSTM网络的论文,并且正在使用PyTorch库。但是我正在努力解决归档训练的概念问题models

使问题更清楚;我将模型保存在一个架构中,我可以以这种形式/models/model-with-loss-2.634221为例。但是对于这种形式,很难确定哪个是哪个。我尝试使用更详细的形式,例如1-layered-100-epoch-128-batchsize-...-etc,但也很难阅读和确定。

您认为处理此类操作最有效的方式是什么?

顺便说一句,我不确定这是正确的网络问这个问题,如果不是,你可以发表评论。

1个回答

一种选择是给每个模型一个唯一的标识符(例如,哈希值或昵称)。然后将所有元数据存储在另一个文件中。

另一种选择是使用 PyTorch 的torch-model-archiver功能。