chansung commited on
Commit
f097832
1 Parent(s): 3593871

Model save

Browse files
README.md ADDED
@@ -0,0 +1,83 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: gemma
3
+ library_name: peft
4
+ tags:
5
+ - trl
6
+ - sft
7
+ - generated_from_trainer
8
+ base_model: google/gemma-2b
9
+ datasets:
10
+ - generator
11
+ model-index:
12
+ - name: gemma2b-summarize-gemini1_5flash-256k
13
+ results: []
14
+ ---
15
+
16
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
17
+ should probably proofread and complete it, then remove this comment. -->
18
+
19
+ # gemma2b-summarize-gemini1_5flash-256k
20
+
21
+ This model is a fine-tuned version of [google/gemma-2b](https://huggingface.co/google/gemma-2b) on the generator dataset.
22
+ It achieves the following results on the evaluation set:
23
+ - Loss: 2.5038
24
+
25
+ ## Model description
26
+
27
+ More information needed
28
+
29
+ ## Intended uses & limitations
30
+
31
+ More information needed
32
+
33
+ ## Training and evaluation data
34
+
35
+ More information needed
36
+
37
+ ## Training procedure
38
+
39
+ ### Training hyperparameters
40
+
41
+ The following hyperparameters were used during training:
42
+ - learning_rate: 0.0002
43
+ - train_batch_size: 8
44
+ - eval_batch_size: 8
45
+ - seed: 42
46
+ - distributed_type: multi-GPU
47
+ - num_devices: 4
48
+ - gradient_accumulation_steps: 2
49
+ - total_train_batch_size: 64
50
+ - total_eval_batch_size: 32
51
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
52
+ - lr_scheduler_type: cosine
53
+ - lr_scheduler_warmup_ratio: 0.1
54
+ - num_epochs: 15
55
+
56
+ ### Training results
57
+
58
+ | Training Loss | Epoch | Step | Validation Loss |
59
+ |:-------------:|:-------:|:----:|:---------------:|
60
+ | 1.0614 | 0.9988 | 414 | 2.4760 |
61
+ | 1.0004 | 2.0 | 829 | 2.4481 |
62
+ | 0.9586 | 2.9988 | 1243 | 2.4426 |
63
+ | 0.9412 | 4.0 | 1658 | 2.4496 |
64
+ | 0.9325 | 4.9988 | 2072 | 2.4600 |
65
+ | 0.9129 | 6.0 | 2487 | 2.4629 |
66
+ | 0.8995 | 6.9988 | 2901 | 2.4703 |
67
+ | 0.8999 | 8.0 | 3316 | 2.4830 |
68
+ | 0.8762 | 8.9988 | 3730 | 2.4934 |
69
+ | 0.8821 | 10.0 | 4145 | 2.4974 |
70
+ | 0.8697 | 10.9988 | 4559 | 2.5013 |
71
+ | 0.8729 | 12.0 | 4974 | 2.5031 |
72
+ | 0.8779 | 12.9988 | 5388 | 2.5023 |
73
+ | 0.8743 | 14.0 | 5803 | 2.5033 |
74
+ | 0.8746 | 14.9819 | 6210 | 2.5038 |
75
+
76
+
77
+ ### Framework versions
78
+
79
+ - PEFT 0.11.1
80
+ - Transformers 4.40.1
81
+ - Pytorch 2.2.0+cu121
82
+ - Datasets 2.19.2
83
+ - Tokenizers 0.19.1
adapter_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:633b70d386a7850b0a336a0cb5c965246328188fe945f7db077d866aea849071
3
  size 19644912
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e024dd42370745139d905182e9d9e06c7c9db5e0335dea5b6af4f1da1146f93f
3
  size 19644912
all_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 14.981905910735826,
3
+ "total_flos": 4.863451355047526e+18,
4
+ "train_loss": 0.9534509487582098,
5
+ "train_runtime": 21285.3674,
6
+ "train_samples": 253412,
7
+ "train_samples_per_second": 18.683,
8
+ "train_steps_per_second": 0.292
9
+ }
runs/Jun05_10-10-03_7bdd13775218/events.out.tfevents.1717582519.7bdd13775218.245062.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6c1d1b72010f4646a9900cde7bb17346fd8a392742a9c5b0b0090d9c3c96eb24
3
- size 270996
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f417fcbd1432ffeb25fb00c81a4579cefa4fa686a342bb2ada1d5f5e889cece0
3
+ size 272043
train_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 14.981905910735826,
3
+ "total_flos": 4.863451355047526e+18,
4
+ "train_loss": 0.9534509487582098,
5
+ "train_runtime": 21285.3674,
6
+ "train_samples": 253412,
7
+ "train_samples_per_second": 18.683,
8
+ "train_steps_per_second": 0.292
9
+ }
trainer_state.json ADDED
The diff for this file is too large to render. See raw diff