level 1
楼主使用的pytorch版本是0.4.1,下面附上源码:
import torch
import torch.utils.data as Data
torch.manual_seed(1) # 设定随机数种子
BATCH_SIZE = 5 # 每批训练的个数
x = torch.linspace(1, 10, 10)
y = torch.linspace(10, 1, 10)
torch_dataset = Data.TensorDataset(x, y)
loader = Data.DataLoader(
dataset=torch_dataset,
batch_size=BATCH_SIZE,
shuffle=True,
num_workers=2,
)
for epoch in range(3):
for step, (batch_x, batch_y) in enumerate(loader):
# training...
print('Epoch: ', epoch, '| Step: ', step, '| batch x: ',
batch_x.numpy(), '| batch y: ', batch_y.numpy())
据说是更新后没有了data_tensor和target_tensor,但是我不会解决啊,求助各位大佬!!!
2018年10月05日 07点10分
2
level 1
将#num_workers=2,去掉即可,
torch_dataset = Data.TensorDataset(x, y) 保留
2018年11月23日 08点11分
4
此方法可行
2019年01月10日 10点01分
为什么呀大佬
2019年04月03日 10点04分
在在在 , 想问一下loader = Data.Dataloader(AttributeError: module 'torch.utils.data' has no attribute 'Dataloader'我搜了好多 没有找到这个原因,这个是因为pytorch 的版本的原因吗?? 我前两天开始看莫烦大神的视频开始零学的
2019年04月13日 05点04分
level 1
按提示加上
if __name__ == '__main__'
应该可以的
2019年02月21日 23点02分
5
level 1
你写线程Num_workers, 那必须得有主线程, 把代码放入 if __name__ == '__main__': 或者把Num_workers这个参数去掉
2019年05月22日 03点05分
6