orendar commited on
Commit
2fa7238
1 Parent(s): f82cf38
Files changed (2) hide show
  1. config.json +20 -34
  2. pytorch_model.bin +2 -2
config.json CHANGED
@@ -1,60 +1,46 @@
1
  {
2
- "_name_or_path": "/home/ec2-user/SageMaker/marian_large",
3
  "activation_dropout": 0.0,
4
- "activation_function": "swish",
5
- "add_bias_logits": false,
6
- "add_final_layer_norm": false,
7
  "architectures": [
8
  "MarianMTModel"
9
  ],
10
  "attention_dropout": 0.0,
11
  "bad_words_ids": [
12
  [
13
- 62954
14
  ]
15
  ],
16
- "bos_token_id": 0,
17
  "classifier_dropout": 0.0,
18
- "d_model": 768,
19
- "decoder_attention_heads": 12,
20
  "decoder_ffn_dim": 4096,
21
  "decoder_layerdrop": 0.0,
22
- "decoder_layers": 9,
23
- "decoder_start_token_id": 62954,
24
- "do_blenderbot_90_layernorm": false,
25
  "dropout": 0.1,
26
- "encoder_attention_heads": 12,
27
  "encoder_ffn_dim": 4096,
28
  "encoder_layerdrop": 0.0,
29
- "encoder_layers": 9,
30
- "eos_token_id": 0,
31
- "extra_pos_embeddings": 2,
32
- "force_bos_token_to_be_generated": false,
33
- "forced_eos_token_id": 0,
34
- "id2label": {
35
- "0": "LABEL_0",
36
- "1": "LABEL_1",
37
- "2": "LABEL_2"
38
- },
39
  "init_std": 0.02,
40
  "is_encoder_decoder": true,
41
- "label2id": {
42
- "LABEL_0": 0,
43
- "LABEL_1": 1,
44
- "LABEL_2": 2
45
- },
46
  "max_length": 512,
47
- "max_position_embeddings": 512,
48
  "model_type": "marian",
49
- "normalize_before": false,
50
  "normalize_embedding": false,
51
- "num_beams": 6,
52
- "num_hidden_layers": 9,
53
- "pad_token_id": 62954,
54
  "scale_embedding": true,
 
55
  "static_position_embeddings": true,
56
  "torch_dtype": "float32",
57
- "transformers_version": "4.12.0.dev0",
58
  "use_cache": true,
59
- "vocab_size": 62955
60
  }
 
1
  {
2
+ "_name_or_path": "output/marian_large_bt_finetuned",
3
  "activation_dropout": 0.0,
4
+ "activation_function": "relu",
 
 
5
  "architectures": [
6
  "MarianMTModel"
7
  ],
8
  "attention_dropout": 0.0,
9
  "bad_words_ids": [
10
  [
11
+ 3
12
  ]
13
  ],
14
+ "bos_token_id": 1,
15
  "classifier_dropout": 0.0,
16
+ "d_model": 1024,
17
+ "decoder_attention_heads": 16,
18
  "decoder_ffn_dim": 4096,
19
  "decoder_layerdrop": 0.0,
20
+ "decoder_layers": 6,
21
+ "decoder_start_token_id": 2,
22
+ "decoder_vocab_size": 32000,
23
  "dropout": 0.1,
24
+ "encoder_attention_heads": 16,
25
  "encoder_ffn_dim": 4096,
26
  "encoder_layerdrop": 0.0,
27
+ "encoder_layers": 6,
28
+ "eos_token_id": 2,
29
+ "forced_eos_token_id": 2,
 
 
 
 
 
 
 
30
  "init_std": 0.02,
31
  "is_encoder_decoder": true,
 
 
 
 
 
32
  "max_length": 512,
33
+ "max_position_embeddings": 1024,
34
  "model_type": "marian",
 
35
  "normalize_embedding": false,
36
+ "num_beams": 4,
37
+ "num_hidden_layers": 6,
38
+ "pad_token_id": 3,
39
  "scale_embedding": true,
40
+ "share_encoder_decoder_embeddings": true,
41
  "static_position_embeddings": true,
42
  "torch_dtype": "float32",
43
+ "transformers_version": "4.19.0.dev0",
44
  "use_cache": true,
45
+ "vocab_size": 32000
46
  }
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:dced33fde86920378b4a0189ee91983072e2d712154c1bc15dabc9463ea4dfa4
3
- size 902523465
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f41589ae763f571855cfba7e1da8198054e0e632aeb5083fec7832c1c2c2916e
3
+ size 836716765