Pytorch
文章平均质量分 65
蜡笔小新不怕辣
RSer!佛系更新佛系回复
展开
-
PyTorch 学习笔记01 - 20210611
一、model.train()和model.eval()用法和区别 例如:定义一个网络: # 线性网络 class Net(nn.Module): def __init__(self): super(generator, self).__init__() self.gen = nn.Sequential( nn.Linear(100, 256), nn.ReLU(True), nn.Linear(原创 2021-06-11 18:05:22 · 259 阅读 · 2 评论 -
Pytorch学习笔记02 - 20210613
模型的保存与加载 一、保存 (一)、torch.save() 和 torch.load() 1. 保存整个模型: # save torch.save(moedl,'save_model.pth') # 调用 model = torch.load('save_model.pth') 2. 仅保存模型参数 # 保存 torch.save(the_model.state_dict(), PATH='model_parameters.pth') #只保存模型权重参数,不保存模型结构 # 调用 the_model原创 2021-06-13 17:10:30 · 223 阅读 · 0 评论