히비스서커스의 블로그

상황 torch로 segmentation model을 돌리는 상황에서 발생하였다. 대략적인 코드 # import unet from seg_model.py from seg_model import unet import torch # output 3 classes model = unet(class=3) # loss 4 classes weights = torch.tensor([1.2, 2.6, 7.5, 17.0], dtype=torch.float32) loss = torch.nn.CrossEntropyLoss(weight=weights) 에러메시지 RuntimeError: only batches of spatial targets supported (3D tensors) but got targets of size..