我用camvid数据集来训练drn时，显示分类权重错误 #55
Description
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