File size: 1,178 Bytes
2ca4c6c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 |
name: trocr_image2text
config_type: model
encoder:
name: trocr_encoder
config_type: model
hidden_size: 768
num_hidden_layers: 12
num_attention_heads: 12
intermediate_size: 3072
hidden_act: gelu
hidden_dropout_prob: 0.0
attention_probs_dropout_prob: 0.0
initializer_range: 0.02
layer_norm_eps: 1.0e-12
image_size: 224
patch_size: 16
num_channels: 3
qkv_bias: true
encoder_stride: 16
decoder:
name: trocr_decoder
config_type: model
is_decoder: true
add_cross_attention: true
attention_probs_dropout_prob: 0.1
bos_token_id: 0
eos_token_id: 2
classifier_dropout: null
gradient_checkpointing: false
hidden_act: gelu
hidden_dropout_prob: 0.1
hidden_size: 768
initializer_range: 0.02
intermediate_size: 3072
layer_norm_eps: 1.0e-12
max_position_embeddings: 514
num_attention_heads: 12
num_hidden_layers: 12
pad_token_id: 1
position_embedding_type: absolute
type_vocab_size: 1
use_cache: true
vocab_size: 42000
generation:
bos_token_id: 0
decoder_start_token_id: 0
early_stopping: true
eos_token_id: 2
length_penalty: 2.0
max_length: 64
no_repeat_ngram_size: 3
num_beams: 4
pad_token_id: 1
|