读写文件


文档摘要

读写文件 到目前为止,我们讨论了如何处理数据, 以及如何构建、训练和测试深度学习模型。 然而,有时我们希望保存训练的模型, 以备将来在各种环境中使用(比如在部署中进行预测)。 此外,当运行一个耗时较长的训练过程时, 最佳的做法是定期保存中间结果, 以确保在服务器电源被不小心断掉时,我们不会损失几天的计算结果。 因此,现在是时候学习如何加载和存储权重向量和整个模型了。 (加载和保存张量) 对于单个张量,我们可以直接调用 和 函数分别读写它们。 这两个函数都要求我们提供一个名称, 要求将要保存的变量作为输入。 我们现在可以将存储在文件中的数据读回内存。 我们可以[存储一个张量列表,然后把它们读回内存。] 我们甚至可以(写入或读取从字符串映射到张量的字典)。


发布者: 作者: 转发
评论区 (0)
U