teodordu commited on
Commit
732ed89
1 Parent(s): dc1ad5c

Added all files from model checkpoint-62500 for vocab_size 20000, 10 epochs

Browse files
config.json ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "/tmp/Helsinki-NLP/opus-mt-en-ro",
3
+ "_num_labels": 3,
4
+ "activation_dropout": 0.0,
5
+ "activation_function": "swish",
6
+ "add_bias_logits": false,
7
+ "add_final_layer_norm": false,
8
+ "architectures": [
9
+ "MarianMTModel"
10
+ ],
11
+ "attention_dropout": 0.0,
12
+ "bad_words_ids": [
13
+ [
14
+ 0
15
+ ]
16
+ ],
17
+ "bos_token_id": 1,
18
+ "classif_dropout": 0.0,
19
+ "classifier_dropout": 0.0,
20
+ "d_model": 512,
21
+ "decoder_attention_heads": 8,
22
+ "decoder_ffn_dim": 2048,
23
+ "decoder_layerdrop": 0.0,
24
+ "decoder_layers": 6,
25
+ "decoder_start_token_id": 3,
26
+ "decoder_vocab_size": 20001,
27
+ "dropout": 0.1,
28
+ "encoder_attention_heads": 8,
29
+ "encoder_ffn_dim": 2048,
30
+ "encoder_layerdrop": 0.0,
31
+ "encoder_layers": 6,
32
+ "eos_token_id": 2,
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": 4,
52
+ "num_hidden_layers": 6,
53
+ "pad_token_id": 3,
54
+ "scale_embedding": true,
55
+ "share_encoder_decoder_embeddings": true,
56
+ "static_position_embeddings": true,
57
+ "torch_dtype": "float32",
58
+ "transformers_version": "4.26.0",
59
+ "unk_token_id": 0,
60
+ "use_cache": true,
61
+ "vocab_size": 20001
62
+ }
generation_config.json ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bad_words_ids": [
4
+ [
5
+ 0
6
+ ]
7
+ ],
8
+ "bos_token_id": 1,
9
+ "decoder_start_token_id": 3,
10
+ "eos_token_id": 2,
11
+ "forced_eos_token_id": 0,
12
+ "max_length": 512,
13
+ "num_beams": 4,
14
+ "pad_token_id": 3,
15
+ "transformers_version": "4.26.0"
16
+ }
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b09aaaeb48e0f8474aaeb8b472917c005ead044310bae8fd1b145b0a17d6b2f1
3
+ size 435180549
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0c47868b626fe48cbaca8248d55c79010e5d8de5788b92e7dc89e4d9f2b5b8a7
3
+ size 217682565
rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1321806b2dca78c343dd16f11f0f25263239484764ca32b5c309b4388a31dd0b
3
+ size 14575
scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e68668651c60748eaaec882baa4b1f243e000fb9a37a9c5f0fa89d100fa1a1df
3
+ size 557
scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:571cbe705e7b8a55ee94dc123b0b79849bcbb95e812f8eba2ef4b4bd7b88dc2f
3
+ size 627
source.spm ADDED
Binary file (617 kB). View file
 
special_tokens_map.json ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {
2
+ "eos_token": "</s>",
3
+ "pad_token": "<pad>",
4
+ "unk_token": "<unk>"
5
+ }
target.spm ADDED
Binary file (617 kB). View file
 
tokenizer_config.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "eos_token": "</s>",
3
+ "model_max_length": 512,
4
+ "name_or_path": "marian-mt-hu-ro-20000",
5
+ "pad_token": "<pad>",
6
+ "separate_vocabs": false,
7
+ "source_lang": null,
8
+ "sp_model_kwargs": {},
9
+ "special_tokens_map_file": null,
10
+ "target_lang": null,
11
+ "tokenizer_class": "MarianTokenizer",
12
+ "unk_token": "<unk>"
13
+ }
trainer_state.json ADDED
@@ -0,0 +1,76 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 10.0,
5
+ "global_step": 62500,
6
+ "is_hyper_param_search": false,
7
+ "is_local_process_zero": true,
8
+ "is_world_process_zero": true,
9
+ "log_history": [
10
+ {
11
+ "epoch": 1.0,
12
+ "learning_rate": 1.9333333333333333e-05,
13
+ "loss": 5.4183,
14
+ "step": 6250
15
+ },
16
+ {
17
+ "epoch": 2.0,
18
+ "learning_rate": 1.866688e-05,
19
+ "loss": 4.1174,
20
+ "step": 12500
21
+ },
22
+ {
23
+ "epoch": 3.0,
24
+ "learning_rate": 1.8000533333333335e-05,
25
+ "loss": 3.5049,
26
+ "step": 18750
27
+ },
28
+ {
29
+ "epoch": 4.0,
30
+ "learning_rate": 1.7334080000000003e-05,
31
+ "loss": 3.0671,
32
+ "step": 25000
33
+ },
34
+ {
35
+ "epoch": 5.0,
36
+ "learning_rate": 1.6667626666666667e-05,
37
+ "loss": 2.7197,
38
+ "step": 31250
39
+ },
40
+ {
41
+ "epoch": 6.0,
42
+ "learning_rate": 1.600128e-05,
43
+ "loss": 2.4474,
44
+ "step": 37500
45
+ },
46
+ {
47
+ "epoch": 7.0,
48
+ "learning_rate": 1.533482666666667e-05,
49
+ "loss": 2.2439,
50
+ "step": 43750
51
+ },
52
+ {
53
+ "epoch": 8.0,
54
+ "learning_rate": 1.4668373333333334e-05,
55
+ "loss": 2.091,
56
+ "step": 50000
57
+ },
58
+ {
59
+ "epoch": 9.0,
60
+ "learning_rate": 1.4001920000000002e-05,
61
+ "loss": 1.9731,
62
+ "step": 56250
63
+ },
64
+ {
65
+ "epoch": 10.0,
66
+ "learning_rate": 1.3335360000000002e-05,
67
+ "loss": 1.8787,
68
+ "step": 62500
69
+ }
70
+ ],
71
+ "max_steps": 187500,
72
+ "num_train_epochs": 30,
73
+ "total_flos": 1.1680685027544269e+17,
74
+ "trial_name": null,
75
+ "trial_params": null
76
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:664e6fb97ea1712a9e07495fe2f23263e6753d0f1182959510e60619ab45de44
3
+ size 3643
vocab.json ADDED
The diff for this file is too large to render. See raw diff