expected target size
-
[이거 어떡하지] Expected target size (, ), got torch.Size([, ]) (nn.CrossEntropyLoss 에서)Pytorch 2021. 6. 18. 22:23
증상: ValueError: Expected target size(256, 84), got torch.Size([256, 64]) image classification이나 language model을 학습시키는 과정에서 nn.CrossEntropyLoss 를 사용하실 때 위와 같은 에러가 발생할 때가 있습니다. (특히 language model처럼 batch 한 샘플마다 sequence별로 classification을 해야 하는 경우) sequence sample별로 output으로 num_classes만큼의 dimension이 뽑히게 냈는데 dimension이 안맞으면 에러가 발생할 수 있습니다. language model의 경우에는 입력으로 [batch_size, len_input] 의 shape를 ..