End of training

README.md

@@ -0,0 +1,75 @@
+---
+license: mit
+base_model: openai-community/gpt2-medium
+tags:
+- trl
+- reward-trainer
+- generated_from_trainer
+metrics:
+- accuracy
+model-index:
+- name: 2c2-reward-medium
+  results: []
+---
+
+<!-- This model card has been generated automatically according to the information the Trainer had access to. You
+should probably proofread and complete it, then remove this comment. -->
+
+# 2c2-reward-medium
+
+This model is a fine-tuned version of [openai-community/gpt2-medium](https://huggingface.co/openai-community/gpt2-medium) on an unknown dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.4456
+- Accuracy: 0.7936
+
+## Model description
+
+More information needed
+
+## Intended uses & limitations
+
+More information needed
+
+## Training and evaluation data
+
+More information needed
+
+## Training procedure
+
+### Training hyperparameters
+
+The following hyperparameters were used during training:
+- learning_rate: 1.41e-05
+- train_batch_size: 256
+- eval_batch_size: 256
+- seed: 42
+- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
+- lr_scheduler_type: constant_with_warmup
+- lr_scheduler_warmup_steps: 20
+- num_epochs: 3.0
+
+### Training results
+
+| Training Loss | Epoch  | Step | Validation Loss | Accuracy |
+|:-------------:|:------:|:----:|:---------------:|:--------:|
+| 1.1138        | 0.2151 | 20   | 1.0849          | 0.5157   |
+| 0.704         | 0.4301 | 40   | 0.7146          | 0.5622   |
+| 0.5512        | 0.6452 | 60   | 0.5744          | 0.7084   |
+| 0.5099        | 0.8602 | 80   | 0.5031          | 0.7478   |
+| 0.5035        | 1.0753 | 100  | 0.4814          | 0.7639   |
+| 0.4409        | 1.2903 | 120  | 0.4687          | 0.7671   |
+| 0.4318        | 1.5054 | 140  | 0.4550          | 0.7831   |
+| 0.4379        | 1.7204 | 160  | 0.4463          | 0.7871   |
+| 0.4402        | 1.9355 | 180  | 0.4366          | 0.7863   |
+| 0.3794        | 2.1505 | 200  | 0.4592          | 0.7936   |
+| 0.3813        | 2.3656 | 220  | 0.4481          | 0.7920   |
+| 0.3634        | 2.5806 | 240  | 0.4384          | 0.8016   |
+| 0.3867        | 2.7957 | 260  | 0.4456          | 0.7936   |
+
+
+### Framework versions
+
+- Transformers 4.41.2
+- Pytorch 2.1.2
+- Datasets 2.19.2
+- Tokenizers 0.19.1

config.json

@@ -0,0 +1,48 @@
+{
+  "_name_or_path": "openai-community/gpt2-medium",
+  "activation_function": "gelu_new",
+  "architectures": [
+    "GPT2ForSequenceClassification"
+  ],
+  "attn_pdrop": 0.1,
+  "bos_token_id": 50256,
+  "embd_pdrop": 0.1,
+  "eos_token_id": 50256,
+  "id2label": {
+    "0": "LABEL_0"
+  },
+  "initializer_range": 0.02,
+  "label2id": {
+    "LABEL_0": 0
+  },
+  "layer_norm_epsilon": 1e-05,
+  "model_type": "gpt2",
+  "n_ctx": 1024,
+  "n_embd": 1024,
+  "n_head": 16,
+  "n_inner": null,
+  "n_layer": 24,
+  "n_positions": 1024,
+  "n_special": 0,
+  "pad_token_id": 50256,
+  "predict_special_tokens": true,
+  "reorder_and_upcast_attn": false,
+  "resid_pdrop": 0.1,
+  "scale_attn_by_inverse_layer_idx": false,
+  "scale_attn_weights": true,
+  "summary_activation": null,
+  "summary_first_dropout": 0.1,
+  "summary_proj_to_labels": true,
+  "summary_type": "cls_index",
+  "summary_use_proj": true,
+  "task_specific_params": {
+    "text-generation": {
+      "do_sample": true,
+      "max_length": 50
+    }
+  },
+  "torch_dtype": "float32",
+  "transformers_version": "4.41.2",
+  "use_cache": true,
+  "vocab_size": 50257
+}

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cae1ddc6d6fe3fa1caf016a922177757b79f5d0ccaaa8ff60d97777e7432d294
+size 1419327048

special_tokens_map.json

@@ -0,0 +1,6 @@
+{
+  "bos_token": "<|endoftext|>",
+  "eos_token": "<|endoftext|>",
+  "pad_token": "<|endoftext|>",
+  "unk_token": "<|endoftext|>"
+}

tokenizer_config.json

@@ -0,0 +1,20 @@
+{
+  "add_prefix_space": false,
+  "added_tokens_decoder": {
+    "50256": {
+      "content": "<|endoftext|>",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "bos_token": "<|endoftext|>",
+  "clean_up_tokenization_spaces": true,
+  "eos_token": "<|endoftext|>",
+  "model_max_length": 1024,
+  "pad_token": "<|endoftext|>",
+  "tokenizer_class": "GPT2Tokenizer",
+  "unk_token": "<|endoftext|>"
+}

training_args.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c23f2a1c2ae4711e3337eeb74bfebed8817230e783cf3af8b16de2eea6e8555c
+size 5176