HarshalH/SFT_LoRA

README.md

@@ -3,9 +3,11 @@ library_name: transformers
 license: mit
 base_model: gpt2
 tags:
+- trl
+- sft
 - generated_from_trainer
 datasets:
-- scitldr
+- generator
 model-index:
 - name: output_dir
   results: []
@@ -16,9 +18,9 @@ should probably proofread and complete it, then remove this comment. -->
 
 # output_dir
 
-This model is a fine-tuned version of [gpt2](https://huggingface.co/gpt2) on the scitldr dataset.
+This model is a fine-tuned version of [gpt2](https://huggingface.co/gpt2) on the generator dataset.
 It achieves the following results on the evaluation set:
-- Loss: 3.3648
+- Loss: 2.4460
 
 ## Model description
 
@@ -37,21 +39,16 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 2e-05
-- train_batch_size: 16
+- learning_rate: 0.0002
+- train_batch_size: 8
 - eval_batch_size: 16
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- num_epochs: 3.0
+- num_epochs: 1
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss |
-|:-------------:|:-----:|:----:|:---------------:|
-| No log        | 1.0   | 125  | 3.4195          |
-| No log        | 2.0   | 250  | 3.3752          |
-| No log        | 3.0   | 375  | 3.3648          |
 
 
 ### Framework versions

model.safetensors

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

runs/Oct06_04-31-13_40898b2f87bd/events.out.tfevents.1728189094.40898b2f87bd.30.2

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

runs/Oct06_04-31-13_40898b2f87bd/events.out.tfevents.1728189247.40898b2f87bd.30.3

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

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ed9f52ffddeb48fb2940c27b404e8f16eed53e637af8752e5a7ac9eb6df99acc
-size 5240
+oid sha256:91370ff73e73b713a06988e65214ddb47583cf6a572fd168afed4c489f69f1d7
+size 5496