sehilnlf commited on
Commit
4b9fdca
1 Parent(s): 478e366

Training in progress, step 500

Browse files
config.json CHANGED
@@ -1,15 +1,15 @@
1
  {
2
- "_name_or_path": "facebook/bart-large-mnli",
3
- "_num_labels": 3,
4
- "activation_dropout": 0.0,
5
  "activation_function": "gelu",
 
6
  "add_final_layer_norm": false,
7
  "architectures": [
8
  "BartForConditionalGeneration"
9
  ],
10
- "attention_dropout": 0.0,
11
  "bos_token_id": 0,
12
- "classif_dropout": 0.0,
13
  "classifier_dropout": 0.0,
14
  "d_model": 1024,
15
  "decoder_attention_heads": 16,
@@ -18,32 +18,55 @@
18
  "decoder_layers": 12,
19
  "decoder_start_token_id": 2,
20
  "dropout": 0.1,
 
21
  "encoder_attention_heads": 16,
22
  "encoder_ffn_dim": 4096,
23
  "encoder_layerdrop": 0.0,
24
  "encoder_layers": 12,
25
  "eos_token_id": 2,
 
26
  "forced_eos_token_id": 2,
27
  "gradient_checkpointing": false,
28
  "id2label": {
29
- "0": "contradiction",
30
- "1": "neutral",
31
- "2": "entailment"
32
  },
33
  "init_std": 0.02,
34
  "is_encoder_decoder": true,
35
  "label2id": {
36
- "contradiction": 0,
37
- "entailment": 2,
38
- "neutral": 1
39
  },
40
  "max_position_embeddings": 1024,
41
  "model_type": "bart",
 
42
  "normalize_before": false,
 
43
  "num_hidden_layers": 12,
44
- "output_past": false,
45
  "pad_token_id": 1,
46
  "scale_embedding": false,
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
47
  "torch_dtype": "float32",
48
  "transformers_version": "4.39.3",
49
  "use_cache": true,
 
1
  {
2
+ "_name_or_path": "facebook/bart-large",
3
+ "activation_dropout": 0.1,
 
4
  "activation_function": "gelu",
5
+ "add_bias_logits": false,
6
  "add_final_layer_norm": false,
7
  "architectures": [
8
  "BartForConditionalGeneration"
9
  ],
10
+ "attention_dropout": 0.1,
11
  "bos_token_id": 0,
12
+ "classif_dropout": 0.1,
13
  "classifier_dropout": 0.0,
14
  "d_model": 1024,
15
  "decoder_attention_heads": 16,
 
18
  "decoder_layers": 12,
19
  "decoder_start_token_id": 2,
20
  "dropout": 0.1,
21
+ "early_stopping": true,
22
  "encoder_attention_heads": 16,
23
  "encoder_ffn_dim": 4096,
24
  "encoder_layerdrop": 0.0,
25
  "encoder_layers": 12,
26
  "eos_token_id": 2,
27
+ "forced_bos_token_id": 0,
28
  "forced_eos_token_id": 2,
29
  "gradient_checkpointing": false,
30
  "id2label": {
31
+ "0": "LABEL_0",
32
+ "1": "LABEL_1",
33
+ "2": "LABEL_2"
34
  },
35
  "init_std": 0.02,
36
  "is_encoder_decoder": true,
37
  "label2id": {
38
+ "LABEL_0": 0,
39
+ "LABEL_1": 1,
40
+ "LABEL_2": 2
41
  },
42
  "max_position_embeddings": 1024,
43
  "model_type": "bart",
44
+ "no_repeat_ngram_size": 3,
45
  "normalize_before": false,
46
+ "num_beams": 4,
47
  "num_hidden_layers": 12,
 
48
  "pad_token_id": 1,
49
  "scale_embedding": false,
50
+ "task_specific_params": {
51
+ "summarization": {
52
+ "length_penalty": 1.0,
53
+ "max_length": 128,
54
+ "min_length": 12,
55
+ "num_beams": 4
56
+ },
57
+ "summarization_cnn": {
58
+ "length_penalty": 2.0,
59
+ "max_length": 142,
60
+ "min_length": 56,
61
+ "num_beams": 4
62
+ },
63
+ "summarization_xsum": {
64
+ "length_penalty": 1.0,
65
+ "max_length": 62,
66
+ "min_length": 11,
67
+ "num_beams": 6
68
+ }
69
+ },
70
  "torch_dtype": "float32",
71
  "transformers_version": "4.39.3",
72
  "use_cache": true,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:17af3ccb39fc52c7cc62a1687ef60962caa542d74e4e13c959b7ec268cc05774
3
  size 1625426996
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f937864fca38a564465f7710e53cac3914427153cae7eaaba92b865ba497d1a6
3
  size 1625426996
runs/May25_06-17-14_332d7128a412/events.out.tfevents.1716617921.332d7128a412.34.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:54388330f8db2bb98d252757fd04e218b21d8d985f8dfaa1bc8848a15773baea
3
+ size 6007
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0fb7b7d5df2f5d5423079772df74162b2945961daccba49eaa4741254b567d88
3
  size 5048
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ab5ccda91aa70574e814c71a849f6934e2bdf40ac505a2726b21265d787b4c65
3
  size 5048