python - 保存和 fast.ai v1 ULMFiT 模型。到/从磁盘

Question

简单的问题。

我一直在用 ULMFiT 做实验

但是我看不到一种方法可以轻松地将模型保存到磁盘或从磁盘加载模型？此示例中创建的 RNNLearner 类有一个可用的加载方法，但没有记录。

我尝试了几种方法，但都没有奏效。有什么我想念的吗？似乎没有从磁盘加载的快速 ai v1 模型的任何示例。

Answer 1

RNNLearner () 类有一个save_encoder()方法。例如，如果您的RNNLearner()对象被调用learn，那么调用调用save_encoder()方法将是：learn.save_encoder(name='my_model_name')

此外，RNNLearner()该类继承自Learner() 类，该类本身包含一个save()方法。

save_encoder()和save()方法都有一个参数name，它接受一个字符串，作为保存模型的名称。