search

guanfuchen / semseg

常用的语义分割架构结构综述以及代码复现 华为媒体研究院 图文Caption、OCR识别、图视文多模态理解与生成相关方向工作或实习欢迎咨询 15757172165 https://guanfuchen.github.io/media/hw_zhaopin_20220724_tiny.jpg
768 stars 164 forks source link

我用camvid数据集来训练drn时，显示分类权重错误 #55

Open SouthernSunset opened 3 years ago

SouthernSunset commented 3 years ago

python train.py --structure drn_d_22 --dataset_path /data/CamVid --cuda True 报错如下：请问我该怎么办 RuntimeError: weight tensor should be defined either for all 512 classes or no classes but got weight tensor of shape: [12] at /opt/conda/conda-bld/pytorch_1579022034529/work/aten/src/THCUNN/generic/ClassNLLCriterion.cu:42