Commit 
							
							·
						
						b2848ac
	
1
								Parent(s):
							
							6dfef1e
								
Upload TFT5ForConditionalGeneration (#1)
Browse files- Upload TFT5ForConditionalGeneration (3c60531743244b8d029e6ef25da3e7a04b1eec3e)
- README.md +57 -0
- config.json +60 -0
- generation_config.json +7 -0
- tf_model.h5 +3 -0
    	
        README.md
    ADDED
    
    | @@ -0,0 +1,57 @@ | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            ---
         | 
| 2 | 
            +
            license: apache-2.0
         | 
| 3 | 
            +
            base_model: t5-base
         | 
| 4 | 
            +
            tags:
         | 
| 5 | 
            +
            - generated_from_keras_callback
         | 
| 6 | 
            +
            model-index:
         | 
| 7 | 
            +
            - name: sum_model
         | 
| 8 | 
            +
              results: []
         | 
| 9 | 
            +
            ---
         | 
| 10 | 
            +
             | 
| 11 | 
            +
            <!-- This model card has been generated automatically according to the information Keras had access to. You should
         | 
| 12 | 
            +
            probably proofread and complete it, then remove this comment. -->
         | 
| 13 | 
            +
             | 
| 14 | 
            +
            # sum_model
         | 
| 15 | 
            +
             | 
| 16 | 
            +
            This model is a fine-tuned version of [t5-base](https://huggingface.co/t5-base) on an unknown dataset.
         | 
| 17 | 
            +
            It achieves the following results on the evaluation set:
         | 
| 18 | 
            +
            - Train Loss: 0.7877
         | 
| 19 | 
            +
            - Validation Loss: 0.5301
         | 
| 20 | 
            +
            - Epoch: 3
         | 
| 21 | 
            +
             | 
| 22 | 
            +
            ## Model description
         | 
| 23 | 
            +
             | 
| 24 | 
            +
            More information needed
         | 
| 25 | 
            +
             | 
| 26 | 
            +
            ## Intended uses & limitations
         | 
| 27 | 
            +
             | 
| 28 | 
            +
            More information needed
         | 
| 29 | 
            +
             | 
| 30 | 
            +
            ## Training and evaluation data
         | 
| 31 | 
            +
             | 
| 32 | 
            +
            More information needed
         | 
| 33 | 
            +
             | 
| 34 | 
            +
            ## Training procedure
         | 
| 35 | 
            +
             | 
| 36 | 
            +
            ### Training hyperparameters
         | 
| 37 | 
            +
             | 
| 38 | 
            +
            The following hyperparameters were used during training:
         | 
| 39 | 
            +
            - optimizer: {'name': 'Adam', 'weight_decay': None, 'clipnorm': None, 'global_clipnorm': None, 'clipvalue': None, 'use_ema': False, 'ema_momentum': 0.99, 'ema_overwrite_frequency': None, 'jit_compile': True, 'is_legacy_optimizer': False, 'learning_rate': 1e-04, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-07, 'amsgrad': False}
         | 
| 40 | 
            +
            - training_precision: float32
         | 
| 41 | 
            +
             | 
| 42 | 
            +
            ### Training results
         | 
| 43 | 
            +
             | 
| 44 | 
            +
            | Train Loss | Validation Loss | Epoch |
         | 
| 45 | 
            +
            |:----------:|:---------------:|:-----:|
         | 
| 46 | 
            +
            | 1.2568     | 0.8907          | 0     |
         | 
| 47 | 
            +
            | 1.0182     | 0.7418          | 1     |
         | 
| 48 | 
            +
            | 0.8905     | 0.6272          | 2     |
         | 
| 49 | 
            +
            | 0.7877     | 0.5301          | 3     |
         | 
| 50 | 
            +
             | 
| 51 | 
            +
             | 
| 52 | 
            +
            ### Framework versions
         | 
| 53 | 
            +
             | 
| 54 | 
            +
            - Transformers 4.32.1
         | 
| 55 | 
            +
            - TensorFlow 2.13.0
         | 
| 56 | 
            +
            - Datasets 2.14.4
         | 
| 57 | 
            +
            - Tokenizers 0.12.1
         | 
    	
        config.json
    ADDED
    
    | @@ -0,0 +1,60 @@ | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            {
         | 
| 2 | 
            +
              "_name_or_path": "t5-base",
         | 
| 3 | 
            +
              "architectures": [
         | 
| 4 | 
            +
                "T5ForConditionalGeneration"
         | 
| 5 | 
            +
              ],
         | 
| 6 | 
            +
              "classifier_dropout": 0.0,
         | 
| 7 | 
            +
              "d_ff": 3072,
         | 
| 8 | 
            +
              "d_kv": 64,
         | 
| 9 | 
            +
              "d_model": 768,
         | 
| 10 | 
            +
              "decoder_start_token_id": 0,
         | 
| 11 | 
            +
              "dense_act_fn": "relu",
         | 
| 12 | 
            +
              "dropout_rate": 0.1,
         | 
| 13 | 
            +
              "eos_token_id": 1,
         | 
| 14 | 
            +
              "feed_forward_proj": "relu",
         | 
| 15 | 
            +
              "initializer_factor": 1.0,
         | 
| 16 | 
            +
              "is_encoder_decoder": true,
         | 
| 17 | 
            +
              "is_gated_act": false,
         | 
| 18 | 
            +
              "layer_norm_epsilon": 1e-06,
         | 
| 19 | 
            +
              "model_type": "t5",
         | 
| 20 | 
            +
              "n_positions": 512,
         | 
| 21 | 
            +
              "num_decoder_layers": 12,
         | 
| 22 | 
            +
              "num_heads": 12,
         | 
| 23 | 
            +
              "num_layers": 12,
         | 
| 24 | 
            +
              "output_past": true,
         | 
| 25 | 
            +
              "pad_token_id": 0,
         | 
| 26 | 
            +
              "relative_attention_max_distance": 128,
         | 
| 27 | 
            +
              "relative_attention_num_buckets": 32,
         | 
| 28 | 
            +
              "task_specific_params": {
         | 
| 29 | 
            +
                "summarization": {
         | 
| 30 | 
            +
                  "early_stopping": true,
         | 
| 31 | 
            +
                  "length_penalty": 2.0,
         | 
| 32 | 
            +
                  "max_length": 200,
         | 
| 33 | 
            +
                  "min_length": 30,
         | 
| 34 | 
            +
                  "no_repeat_ngram_size": 3,
         | 
| 35 | 
            +
                  "num_beams": 4,
         | 
| 36 | 
            +
                  "prefix": "summarize: "
         | 
| 37 | 
            +
                },
         | 
| 38 | 
            +
                "translation_en_to_de": {
         | 
| 39 | 
            +
                  "early_stopping": true,
         | 
| 40 | 
            +
                  "max_length": 300,
         | 
| 41 | 
            +
                  "num_beams": 4,
         | 
| 42 | 
            +
                  "prefix": "translate English to German: "
         | 
| 43 | 
            +
                },
         | 
| 44 | 
            +
                "translation_en_to_fr": {
         | 
| 45 | 
            +
                  "early_stopping": true,
         | 
| 46 | 
            +
                  "max_length": 300,
         | 
| 47 | 
            +
                  "num_beams": 4,
         | 
| 48 | 
            +
                  "prefix": "translate English to French: "
         | 
| 49 | 
            +
                },
         | 
| 50 | 
            +
                "translation_en_to_ro": {
         | 
| 51 | 
            +
                  "early_stopping": true,
         | 
| 52 | 
            +
                  "max_length": 300,
         | 
| 53 | 
            +
                  "num_beams": 4,
         | 
| 54 | 
            +
                  "prefix": "translate English to Romanian: "
         | 
| 55 | 
            +
                }
         | 
| 56 | 
            +
              },
         | 
| 57 | 
            +
              "transformers_version": "4.32.1",
         | 
| 58 | 
            +
              "use_cache": true,
         | 
| 59 | 
            +
              "vocab_size": 32128
         | 
| 60 | 
            +
            }
         | 
    	
        generation_config.json
    ADDED
    
    | @@ -0,0 +1,7 @@ | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            {
         | 
| 2 | 
            +
              "_from_model_config": true,
         | 
| 3 | 
            +
              "decoder_start_token_id": 0,
         | 
| 4 | 
            +
              "eos_token_id": 1,
         | 
| 5 | 
            +
              "pad_token_id": 0,
         | 
| 6 | 
            +
              "transformers_version": "4.32.1"
         | 
| 7 | 
            +
            }
         | 
    	
        tf_model.h5
    ADDED
    
    | @@ -0,0 +1,3 @@ | |
|  | |
|  | |
|  | 
|  | |
| 1 | 
            +
            version https://git-lfs.github.com/spec/v1
         | 
| 2 | 
            +
            oid sha256:0b0cf747ceae73a38d32a3946324f14b7f263e15eee97e455c7f38df28bcdc02
         | 
| 3 | 
            +
            size 1089544048
         |