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