- 博客(2)
- 收藏
- 关注
RSS订阅原创 全连接层与torch.nn.linear()理解
在pytorch中的nn.Linear表示线性变换，官方文档给出的数学计算公式是其中x是输入，A是权值，b是偏置，y是输出，卷积神经网络中的全连接层需要调用nn.Linear就可以实现，其源代码为class Linear(Module): def __init__(self, in_features: int, out_features: int, bias: bool = True) -> None:参数： in_features: size of each in
2020-12-16 14:51:57
34014
原创 torch.save()模型的保存于加载
保存模型主要分为两类：保存整个模型和只保存模型参数1.保存加载整个模型（不推荐）： 保存整个网络模型，网络结构+权重参数torch.save(model,'net.pth')加载整个网络模型（可能比较耗时）model=torch.load('net.pth')2.只保存加载模型参数（推荐） 保存模型的权重参数（速度快，占内存少）torch.save(model.state_dict(),'net_params.pth')load 模型参数因为我们只保存了 ...
2020-12-14 11:54:31
29242
4
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人