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