archit11/qwen-finetuned-model

README.md

@@ -0,0 +1,76 @@
+---
+library_name: transformers
+license: apache-2.0
+base_model: Qwen/Qwen2.5-0.5B
+tags:
+- generated_from_trainer
+model-index:
+- name: capybara_finetuned_results3
+  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. -->
+
+# capybara_finetuned_results3
+
+This model is a fine-tuned version of [Qwen/Qwen2.5-0.5B](https://huggingface.co/Qwen/Qwen2.5-0.5B) on an unknown dataset.
+It achieves the following results on the evaluation set:
+- Loss: 5.6542
+
+## 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: 0.0002
+- train_batch_size: 1
+- eval_batch_size: 8
+- seed: 42
+- gradient_accumulation_steps: 4
+- total_train_batch_size: 4
+- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
+- lr_scheduler_type: cosine
+- lr_scheduler_warmup_steps: 5
+- training_steps: 800
+
+### Training results
+
+| Training Loss | Epoch  | Step | Validation Loss |
+|:-------------:|:------:|:----:|:---------------:|
+| 15.5311       | 0.0230 | 50   | 14.5422         |
+| 8.7477        | 0.0460 | 100  | 9.2952          |
+| 7.3554        | 0.0690 | 150  | 7.1992          |
+| 6.828         | 0.0920 | 200  | 6.7258          |
+| 6.4694        | 0.1150 | 250  | 6.3597          |
+| 6.3401        | 0.1381 | 300  | 6.1703          |
+| 6.1256        | 0.1611 | 350  | 6.0395          |
+| 6.0372        | 0.1841 | 400  | 5.9271          |
+| 6.0221        | 0.2071 | 450  | 5.8464          |
+| 5.8783        | 0.2301 | 500  | 5.7810          |
+| 5.8339        | 0.2531 | 550  | 5.7335          |
+| 5.8546        | 0.2761 | 600  | 5.6904          |
+| 5.9169        | 0.2991 | 650  | 5.6690          |
+| 5.7959        | 0.3221 | 700  | 5.6565          |
+| 5.7271        | 0.3451 | 750  | 5.6543          |
+| 5.8734        | 0.3682 | 800  | 5.6542          |
+
+
+### Framework versions
+
+- Transformers 4.44.2
+- Pytorch 2.4.0
+- Datasets 3.0.0
+- Tokenizers 0.19.1

config.json

@@ -0,0 +1,29 @@
+{
+  "_name_or_path": "Qwen/Qwen2.5-0.5B",
+  "architectures": [
+    "Qwen2ForCausalLM"
+  ],
+  "attention_dropout": 0.0,
+  "bos_token_id": 151643,
+  "eos_token_id": 151643,
+  "hidden_act": "silu",
+  "hidden_size": 896,
+  "initializer_range": 0.02,
+  "intermediate_size": 4864,
+  "max_position_embeddings": 32768,
+  "max_window_layers": 24,
+  "model_type": "qwen2",
+  "num_attention_heads": 14,
+  "num_hidden_layers": 24,
+  "num_key_value_heads": 2,
+  "rms_norm_eps": 1e-06,
+  "rope_theta": 1000000.0,
+  "sliding_window": null,
+  "tie_word_embeddings": true,
+  "torch_dtype": "bfloat16",
+  "transformers_version": "4.44.2",
+  "use_cache": true,
+  "use_mrope": false,
+  "use_sliding_window": false,
+  "vocab_size": 151936
+}

generation_config.json

@@ -0,0 +1,6 @@
+{
+  "bos_token_id": 151643,
+  "eos_token_id": 151643,
+  "max_new_tokens": 2048,
+  "transformers_version": "4.44.2"
+}

model.safetensors

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

runs/Sep28_10-42-26_e89ab5377562/events.out.tfevents.1727520148.e89ab5377562.24.0

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

training_args.bin

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