- 博客(2)
- 收藏
- 关注
原创 torch.utils.data.DataLoader 官方文档学习
Pytorch 首选的数据上传工具是 torch.utils.data.DataLoader,它可以将已经预处理完成的数据集转换成一个迭代器,迭代一次输出一个预先定义好的batch_size数量的图片torch.utils.data.DataLoader类1、torch.utils.data.DataLoader(dataset, batch_size=1, shuffle=False, ...
2019-08-19 14:42:31 1695
原创 Torch.nn.Module 模块官方文档学习
1、nn.parameterTensor类的特殊化实例,利用torch.nn.Parameter(data,required_grad = True)定义的变量可以直接作为网络模型的parameter2、nn.Moduletorch中搭建神经网络的父类add_module(name,module)向当前模块中添加子模块,添加的模块可以通过name访问apply(fn)...
2019-08-18 18:15:20 1318
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人