- 博客(3)
- 收藏
- 关注
RSS订阅原创 Softmax函数与交叉熵损失
在分类问题中，直接使用输出层的输出有两个问题。一方面，由于输出层的输出值的范围不确定，我们难以直观上判断这些值的意义。例如，刚才举的例子中的输出值10表示“很置信”图像类别为猫，因为该输出值是其他两类的输出值的100倍。但如果o1=o3=103o_1=o_3=10^3o1=o3=103，那么输出值10却又表示图像类别为猫的概率很低。另一方面，由于真实标签是离散值，这些离散值与不确定范围的输出值...
2019-09-17 22:38:17
331
原创 《Dive-into-deeplearning》3.2 用Pytorch实现线性回归
读取数据 原文来自： import torch.utils.data as Data batch_size = 10 # 将训练数据的特征和标签组合 dataset = Data.TensorDataset(features, labels) # 随机读取小批量 data_iter = Data.DataLoader(dataset, batch_size, shuffle=True) 定义网...
2019-09-17 22:30:19
318
原创 Pytorh-存储和加载模型参数
参考：pytorch学习笔记（五）：保存和加载模型 #保存和加载模型参数 torch.save(model_object, 'model.pkl') model = torch.load('model.pkl') # 仅保存和加载模型参数(推荐使用) torch.save(model_object.state_dict(), 'params.pkl') model_object.load_sta...
2019-09-04 15:37:05
217
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人