摘要:
Pytorch有两个读取数据的方式: 使用Dataset 使用DataLoader 本文先介绍第一种——Dataset Dataset与DataLoader区别 Dataset:提供一种方法,去获取数据及其对应的label值 DataLoader:提供一种方法,可以以特定的形式打包数据 Refere 阅读全文
摘要:
在训练一个网络过程中,有下面代码 '''4.训练网络''' print('开始训练') for epoch in range(3): runing_loss = 0.0 for i, data in enumerate(trainloader, 0): inputs, label = data # 阅读全文
摘要:
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 阅读全文