pytorch-deeplab-xception icon indicating copy to clipboard operation
pytorch-deeplab-xception copied to clipboard

Published 20 hours ago • verified publisher icon jfzhang95

loss function

Open FANG-MING opened this issue 5 years ago • 1 comments

when i use CrossEntropyLoss(),it has aproblem:RuntimeError: 1only batches of spatial targets supported (non-empty 3D tensors) but got targets of size: : [16, 513, 513, 4]

FANG-MING avatar Oct 18 '20 08:10 FANG-MING

target should be 3 dimension.

lipanpeng avatar Dec 25 '20 03:12 lipanpeng