使用了自定义的dict,发现fc层的维度与词典不对应。请问怎么设置FC层的维度？
v8wbuo2f1#
修改一下对应yml文件中模型的fc层参数
2ic8powd2#
Architecture:model_type: recalgorithm: CRNNTransform:Backbone:name: MobileNetV3scale: 0.5model_name: smallsmall_stride: [1, 2, 2, 2]Neck:name: SequenceEncoderencoder_type: rnnhidden_size: 48Head:name: CTCHeadfc_decay: 0.00001请问应该修改哪个参数？
szqfcxe23#
PaddleOCR/ppocr/modeling/heads/rec_ctc_head.py
Line 38 in 18ddb6d
| | out_channels, |这里
3条答案
按热度按时间v8wbuo2f1#
修改一下对应yml文件中模型的fc层参数
2ic8powd2#
Architecture:
model_type: rec
algorithm: CRNN
Transform:
Backbone:
name: MobileNetV3
scale: 0.5
model_name: small
small_stride: [1, 2, 2, 2]
Neck:
name: SequenceEncoder
encoder_type: rnn
hidden_size: 48
Head:
name: CTCHead
fc_decay: 0.00001
请问应该修改哪个参数？
szqfcxe23#
PaddleOCR/ppocr/modeling/heads/rec_ctc_head.py
Line 38 in 18ddb6d
| | out_channels, |
这里