paddle.fluid.core.EnforceNotMet: Enforce failed. Expected in_dims[1] == filter_dims[1] * groups, but received in_dims[1]:32 != filter_dims[1] * groups:1024.
Created by: imistyrain
使用object_detection在自己的数据集(仅含两类,背景和物体)上进行微调时报错,与预训练的模型类别不一致。而如果不使用预训练模型的话则可以正常训练. 所用命令为
python -u train.py --image_shape=3,$SHAPE,$SHAPE \
--pretrained_model='pretrained/ssd_mobilenet_v1_coco/' \
--model_save_dir=model
有没有类似于caffe中更改层的名字以便重新初始化模型参数的方法来在paddle中对最后这些层重新初始化而不是从预训练模型中加载呢?