0
点赞
收藏
分享

微信扫一扫

关于pytorch在训练模型时常见错误

A邱凌 2022-04-02 阅读 59

文章目录

关于pytorch在训练模型时常见错误

问题1

【Pytorch】Expected hidden[0] size (2, 32, 256), got [2, 14, 256]

错误原因:

改正方法:

  1. 修改batchsize,让数据集大小能整除batchsize
  2. 如果使用Dataloader,设置一个参数drop_last=True,会自动舍弃最后不足batchsize的batch

问题2

1D target tensor expected, multi-target not supported

错误原因:

改正方法:

  1. 查看label和pred的shape形状
pred = rnn(data)	#预测
label.shape			#标签
pred.shape
label.shapepred.shape
(batch,)(batch,classes)
(batch,seq_len)(batch,classes,seq_len)

问题3

too many values to unpack (expected 2,3......)

错误原因:

改正方法:

  1. 对应每一个值

问题4 np转int

IndexError: too many indices for array: array is 0-dimensional, but 1 were indexed解决方案

错误原因:


在这里插入图片描述
当你取没有维度的np时会报错
在这里插入图片描述

改正方法:

在这里插入图片描述

问题5

unhashable type: 'numpy.ndarray'

错误原因:


在这里插入图片描述

改正方法:
在这里插入图片描述

举报

相关推荐

0 条评论