End of training

config.json

@@ -24,6 +24,7 @@
   "decoder_layers": 6,
   "decoder_start_token_id": 2,
   "dropout": 0.1,
+  "early_stopping": true,
   "encoder_attention_heads": 12,
   "encoder_ffn_dim": 3072,
   "encoder_layerdrop": 0.0,
@@ -42,9 +43,14 @@
     "LABEL_1": 1,
     "LABEL_2": 2
   },
+  "length_penalty": 2.0,
   "max_decoder_position_embeddings": 1024,
   "max_encoder_position_embeddings": 16384,
+  "max_length": 16896,
+  "min_length": 100,
   "model_type": "led",
+  "no_repeat_ngram_size": 3,
+  "num_beams": 4,
   "num_hidden_layers": 6,
   "pad_token_id": 1,
   "torch_dtype": "float32",

generation_config.json

@@ -2,7 +2,13 @@
   "_from_model_config": true,
   "bos_token_id": 0,
   "decoder_start_token_id": 2,
+  "early_stopping": true,
   "eos_token_id": 2,
+  "length_penalty": 2.0,
+  "max_length": 16896,
+  "min_length": 100,
+  "no_repeat_ngram_size": 3,
+  "num_beams": 4,
   "pad_token_id": 1,
   "transformers_version": "4.26.0"
 }

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:696da0d2af886c28708a62d7e06ee39c14d10764f9163c421da786db73951352
+oid sha256:3386f24dcb6b390d958be8bf31e64f56f4b32cb13a1cf7d4d5d62f0680593faf
 size 647702381

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:924c2f5b07e2f3daef861d7a30e1544de90a2b5bf0fb0d965cd633233de7bc30
+oid sha256:fe29feb18600443487bbce36543a1d7ed9f62f7e30731f19745c2c8195df0d08
 size 3707