按教程进行预训练报错:“finetune_exclude_pretrained_params”
Created by: LRY0111
按照官方教程(https://github.com/PaddlePaddle/models/tree/develop/PaddleCV/image_classification)进行预训练微调参数,训练自己的数据时需要滤掉最后的FC层报错:”train.py: error: unrecognized arguments: --finetune_exclude_pretrained_params=fc_offset,fc_weights“
错误很明显:官方的train.py代码中没有“finetune_exclude_pretrained_params”参数,请官方人员回复一下,如果没有这个参数,我做微调如何去掉最后一层的参数呢???
1)官方给的命令:
python train.py
--data_dir=./data/ILSVRC2012/
--total_images=1281167
--class_dim=1000
--validate=True
--model=ResNet50_vd
--batch_size=256 \
--lr=0.1
--num_epochs=200
--model_save_dir=output/
--l2_decay=7e-5
--pretrained_model=${path_to_pretrain_model}
--finetune_exclude_pretrained_params=fc_0.w_0,fc_0.b_0
2)我的命令: python train.py --model=DPN107 --data_dir=input_image/train_images --total_images=20580 --class_dim=120 --batch_size=30 --pretrained_model=models_parameters/pretrained_model/DPN107 --model_save_dir=models_parameters/training_model/DPN107 --finetune_exclude_pretrained_params=fc_offset,fc_weights