aephil commited on
Commit
fbdbcfd
·
verified ·
1 Parent(s): 4f56057

End of training

Browse files
Files changed (5) hide show
  1. README.md +68 -0
  2. config.json +4 -6
  3. model.safetensors +2 -2
  4. tokenizer_config.json +1 -1
  5. training_args.bin +1 -1
README.md ADDED
@@ -0,0 +1,68 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: apache-2.0
4
+ base_model: HuggingFaceTB/SmolLM-135M
5
+ tags:
6
+ - generated_from_trainer
7
+ model-index:
8
+ - name: lab7-shakespeare-generator2
9
+ results: []
10
+ ---
11
+
12
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
13
+ should probably proofread and complete it, then remove this comment. -->
14
+
15
+ # lab7-shakespeare-generator2
16
+
17
+ This model is a fine-tuned version of [HuggingFaceTB/SmolLM-135M](https://huggingface.co/HuggingFaceTB/SmolLM-135M) on an unknown dataset.
18
+ It achieves the following results on the evaluation set:
19
+ - Loss: 3.5582
20
+
21
+ ## Model description
22
+
23
+ More information needed
24
+
25
+ ## Intended uses & limitations
26
+
27
+ More information needed
28
+
29
+ ## Training and evaluation data
30
+
31
+ More information needed
32
+
33
+ ## Training procedure
34
+
35
+ ### Training hyperparameters
36
+
37
+ The following hyperparameters were used during training:
38
+ - learning_rate: 5e-05
39
+ - train_batch_size: 4
40
+ - eval_batch_size: 8
41
+ - seed: 42
42
+ - gradient_accumulation_steps: 2
43
+ - total_train_batch_size: 8
44
+ - optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
45
+ - lr_scheduler_type: cosine
46
+ - lr_scheduler_warmup_steps: 50
47
+ - num_epochs: 2
48
+
49
+ ### Training results
50
+
51
+ | Training Loss | Epoch | Step | Validation Loss |
52
+ |:-------------:|:------:|:----:|:---------------:|
53
+ | 3.8418 | 0.2747 | 100 | 3.6673 |
54
+ | 3.7432 | 0.5495 | 200 | 3.6014 |
55
+ | 3.6580 | 0.8242 | 300 | 3.5733 |
56
+ | 3.6421 | 1.0989 | 400 | 3.5628 |
57
+ | 3.6488 | 1.3736 | 500 | 3.5589 |
58
+ | 3.6311 | 1.6484 | 600 | 3.5580 |
59
+ | 3.6232 | 1.9231 | 700 | 3.5579 |
60
+ | 3.6232 | 2.0 | 728 | 3.5582 |
61
+
62
+
63
+ ### Framework versions
64
+
65
+ - Transformers 5.12.1
66
+ - Pytorch 2.12.1
67
+ - Datasets 5.0.0
68
+ - Tokenizers 0.22.2
config.json CHANGED
@@ -5,15 +5,14 @@
5
  "attention_bias": false,
6
  "attention_dropout": 0.0,
7
  "bos_token_id": 0,
8
- "dtype": "float16",
9
  "eos_token_id": 0,
10
  "head_dim": 64,
11
  "hidden_act": "silu",
12
  "hidden_size": 576,
13
- "initializer_range": 0.041666666666666664,
14
  "intermediate_size": 1536,
15
- "is_llama_config": true,
16
- "max_position_embeddings": 8192,
17
  "mlp_bias": false,
18
  "model_type": "llama",
19
  "num_attention_heads": 9,
@@ -22,9 +21,8 @@
22
  "pad_token_id": 0,
23
  "pretraining_tp": 1,
24
  "rms_norm_eps": 1e-05,
25
- "rope_interleaved": false,
26
  "rope_parameters": {
27
- "rope_theta": 100000,
28
  "rope_type": "default"
29
  },
30
  "tie_word_embeddings": true,
 
5
  "attention_bias": false,
6
  "attention_dropout": 0.0,
7
  "bos_token_id": 0,
8
+ "dtype": "bfloat16",
9
  "eos_token_id": 0,
10
  "head_dim": 64,
11
  "hidden_act": "silu",
12
  "hidden_size": 576,
13
+ "initializer_range": 0.02,
14
  "intermediate_size": 1536,
15
+ "max_position_embeddings": 2048,
 
16
  "mlp_bias": false,
17
  "model_type": "llama",
18
  "num_attention_heads": 9,
 
21
  "pad_token_id": 0,
22
  "pretraining_tp": 1,
23
  "rms_norm_eps": 1e-05,
 
24
  "rope_parameters": {
25
+ "rope_theta": 10000.0,
26
  "rope_type": "default"
27
  },
28
  "tie_word_embeddings": true,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3e11b06d6977f560ef57765748dcd17a8437b6d2acf7e9e68c2d8de353e133bd
3
- size 269060280
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e8b97f551ce6a98454cf3d74fd692a4964511814d9a7aa3687d5de1082b37405
3
+ size 269060552
tokenizer_config.json CHANGED
@@ -26,7 +26,7 @@
26
  ],
27
  "is_local": false,
28
  "local_files_only": false,
29
- "model_max_length": 8192,
30
  "pad_token": "<|endoftext|>",
31
  "tokenizer_class": "GPT2Tokenizer",
32
  "unk_token": "<|endoftext|>",
 
26
  ],
27
  "is_local": false,
28
  "local_files_only": false,
29
+ "model_max_length": 1000000000000000019884624838656,
30
  "pad_token": "<|endoftext|>",
31
  "tokenizer_class": "GPT2Tokenizer",
32
  "unk_token": "<|endoftext|>",
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:26f66913e3a60bf02c0595baeef7b68061375d221d3439e9094b7fa0206326a7
3
  size 5137
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5029b829699a7326d00774c0c25903aef5a164c9d371b9570ba098490a6df51b
3
  size 5137