摘要: Pytorch有两个读取数据的方式: 使用Dataset 使用DataLoader 本文先介绍第一种——Dataset Dataset与DataLoader区别 Dataset:提供一种方法,去获取数据及其对应的label值 DataLoader:提供一种方法,可以以特定的形式打包数据 Refere 阅读全文
posted @ 2024-03-19 11:11 光辉233 阅读(14) 评论(0) 推荐(0) 编辑
摘要: 在训练一个网络过程中,有下面代码 '''4.训练网络''' print('开始训练') for epoch in range(3): runing_loss = 0.0 for i, data in enumerate(trainloader, 0): inputs, label = data # 阅读全文
posted @ 2024-03-19 10:59 光辉233 阅读(2) 评论(0) 推荐(0) 编辑
摘要: print('开始训练') for epoch in range(3): runing_loss = 0.0 for i,data in enumerate(trainloader,0): inputs,label = data #1.数据加载 if device == 'gpu': inputs 阅读全文
posted @ 2024-03-19 00:13 光辉233 阅读(4) 评论(0) 推荐(0) 编辑