2018年12月18日

Pytorch基础(5)——批数据训练

摘要: 一、知识点: 相关包:torch.utils.data 相关包:torch.utils.data 包装数据类:TensorDataset 包装数据类:TensorDataset 【包装数据和目标张量的数据集,通过沿着第一个维度索引两个张量来】 加载数据类:DataLoader 加载数据类:DataL 阅读全文

posted @ 2018-12-18 20:21 吱吱了了 阅读(3955) 评论(0) 推荐(0)

pytorch基础(4)-----搭建模型网络的两种方法

摘要: 方法一:采用torch.nn.Module模块 打印的结果: Net( (hidden): Linear(in_features=2, out_features=10, bias=True) (predict): Linear(in_features=10, out_features=2, bias 阅读全文

posted @ 2018-12-18 19:42 吱吱了了 阅读(355) 评论(0) 推荐(0)

Pytorch实战(3)----分类

摘要: 一、分类任务: 将以下两类分开。 创建数据代码: 二、步骤 导入包 创建模型 设置优化器和损失函数 训练模型 三、代码: 导入包: 创建模型: 设置优化器和损失函数 训练模型并画图展示 结果展示: 阅读全文

posted @ 2018-12-18 19:16 吱吱了了 阅读(580) 评论(0) 推荐(0)

导航