PyTorch
`AllureLove
这个作者很懒,什么都没留下…
展开
-
pytorch embedding层报错IndexError: index out of range in self
IndexError: index out of range in self报这个错误是embedding层的张量输入超过了合法范围,embedding层的合法张量输入数值范围应该在[0, num_embeddings - 1]的范围内,过大过小都会报错。因此可以在embedding层先输出查看一下最大范围:print(self.embedding.num_embeddings)然后查看一下输入embedding层数据张量的范围:print(input_tensor.min())print.原创 2021-07-05 16:42:21 · 6661 阅读 · 0 评论 -
Target xxx is out of bounds.
在采用pytorch进行训练时,最后计算cross entropy损失时会报错。全连接层最终的维度num_classes表示的是类别的个数,但是在进行自定义数据集的标签设置时,不能从1开始,而是从0开始。如一共有四类,则标签类别为0,1,2,3。标注范围为[0, num_classes-1],一旦超出这个范围则会报错。在pytorch中,输入数据形式时[batch_size, channel, H, W],在lstm模块中如果不设置batch_first=True,则需要将tensor的batch和se原创 2021-11-03 16:11:35 · 8880 阅读 · 0 评论