- 博客(2)
- 收藏
- 关注
转载 PyTorch学习笔记:EOFError:Ran out of input
报错分析:这个错误其实是pytorch函数torch.utils.data.DataLoader在windows下的特有错误,该函数里面有个参数num_workers表示进程个数,在windows下改为0就可以了解决方法:train_loader = Data.DataLoader( dataset=data_x, # 使用的数据集※ batch_size=64, # 批处理样本大小 shuffle=True, # 每次迭代前打乱数据 ...
2022-02-10 00:50:03 3636 2
原创 交叉熵---RuntimeError: 1D target tensor expected, multi-target not supported
pytorch中计计算交叉熵损失函数时, 输入的正确 label 不能是 one-hot 格式。函数内部会自己处理成 one hot 格式。所以不需要输入 [ 0 0 0 0 1],只需要输入 4 就行。
2022-01-15 04:33:16 920
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人