chansung commited on
Commit
940519c
1 Parent(s): 3f10d61

Model save

Browse files
README.md ADDED
@@ -0,0 +1,69 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: gemma
3
+ library_name: peft
4
+ tags:
5
+ - trl
6
+ - sft
7
+ - generated_from_trainer
8
+ base_model: google/gemma-7b
9
+ datasets:
10
+ - generator
11
+ model-index:
12
+ - name: mental_health_counseling_merged_v0.1
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
+ # mental_health_counseling_merged_v0.1
20
+
21
+ This model is a fine-tuned version of [google/gemma-7b](https://huggingface.co/google/gemma-7b) on the generator dataset.
22
+ It achieves the following results on the evaluation set:
23
+ - Loss: 0.8574
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: 2
44
+ - eval_batch_size: 2
45
+ - seed: 42
46
+ - distributed_type: multi-GPU
47
+ - num_devices: 2
48
+ - gradient_accumulation_steps: 2
49
+ - total_train_batch_size: 8
50
+ - total_eval_batch_size: 4
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: 1
55
+
56
+ ### Training results
57
+
58
+ | Training Loss | Epoch | Step | Validation Loss |
59
+ |:-------------:|:------:|:-----:|:---------------:|
60
+ | 0.8429 | 1.0000 | 10673 | 0.8574 |
61
+
62
+
63
+ ### Framework versions
64
+
65
+ - PEFT 0.11.0
66
+ - Transformers 4.40.2
67
+ - Pytorch 2.3.0+cu121
68
+ - Datasets 2.19.1
69
+ - Tokenizers 0.19.1
adapter_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:bfa5283c1e536643da77ca6b06b7d1cf1b70198641a6f2d07b5c66f04541b6ed
3
  size 100060536
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b8bf97159aba13310795421f202d4859fb345868ed49032622843e570c3ff141
3
  size 100060536
all_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 0.9999531550100716,
3
+ "total_flos": 4.092531867595571e+18,
4
+ "train_loss": 1.0194565681556524,
5
+ "train_runtime": 58622.8041,
6
+ "train_samples": 97468,
7
+ "train_samples_per_second": 1.457,
8
+ "train_steps_per_second": 0.182
9
+ }
runs/May17_03-53-04_instance-20240517-0630/events.out.tfevents.1715932833.instance-20240517-0630.565.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b72efa53917b00a66ff4de181654d78415a9debf651f523593172e1c25d0891f
3
- size 452903
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:00e0d6976876c33ecbf7e6d38e30de3f4c2496bf6a74395bdc1cb3e0de35cc24
3
+ size 456482
train_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 0.9999531550100716,
3
+ "total_flos": 4.092531867595571e+18,
4
+ "train_loss": 1.0194565681556524,
5
+ "train_runtime": 58622.8041,
6
+ "train_samples": 97468,
7
+ "train_samples_per_second": 1.457,
8
+ "train_steps_per_second": 0.182
9
+ }
trainer_state.json ADDED
The diff for this file is too large to render. See raw diff