search

CSAILVision / semantic-segmentation-pytorch

Pytorch implementation for Semantic Segmentation/Scene Parsing on MIT ADE20K dataset
http://sceneparsing.csail.mit.edu/
BSD 3-Clause "New" or "Revised" License
4.96k stars 1.1k forks source link

Training Custom Dataset Error #217

Closed mdt48 closed 4 years ago

mdt48 commented 4 years ago

Hi, I'm trying to train on my own dataset and I keep getting the following error:

/opt/conda/conda-bld/pytorch_1573049304260/work/aten/src/THCUNN/SpatialClassNLLCriterion.cu:104: void cunn_SpatialClassNLLCriterion_updateOutput_kernel(T *, T *, T *, long *, T *, int, int, int, int, int, long) [with T = float, AccumT = float]: block: [0,0,0], thread: [319,0,0] Assertion `t >= 0 && t < n_classes` failed.

Any help would be appreciated! Thanks in advance!

Param-Raval commented 4 years ago

how did you solve this?

ShihuaiXu commented 4 years ago

how did you solve this?