2023年12月8日
在pytorch中保存模型或模型参数
摘要： 在 PyTorch 中，我们可以使用 torch.save 函数将 PyTorch 模型保存到文件。这个函数接受两个参数：要保存的对象（通常是模型），以及文件路径。 保存模型参数 import torch import torch.nn as nn # 假设有一个简单的模型 class Simple 阅读全文
posted @ 2023-12-08 16:56 落魄统计佬 阅读(223) 评论(0) 推荐(0) 编辑