NAM00 commited on
Commit
0c84aa0
·
verified ·
1 Parent(s): cc2594e

Trained patient-matching model with context-tuning

Browse files
Files changed (2) hide show
  1. README.md +8 -7
  2. model.safetensors +1 -1
README.md CHANGED
@@ -16,7 +16,7 @@ should probably proofread and complete it, then remove this comment. -->
16
 
17
  This model is a fine-tuned version of [meta-llama/Llama-3.2-1B-Instruct](https://huggingface.co/meta-llama/Llama-3.2-1B-Instruct) on an unknown dataset.
18
  It achieves the following results on the evaluation set:
19
- - Loss: 1.6590
20
 
21
  ## Model description
22
 
@@ -35,7 +35,7 @@ More information needed
35
  ### Training hyperparameters
36
 
37
  The following hyperparameters were used during training:
38
- - learning_rate: 3e-05
39
  - train_batch_size: 1
40
  - eval_batch_size: 1
41
  - seed: 42
@@ -43,17 +43,18 @@ The following hyperparameters were used during training:
43
  - total_train_batch_size: 8
44
  - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
45
  - lr_scheduler_type: cosine
 
46
  - num_epochs: 5
47
 
48
  ### Training results
49
 
50
  | Training Loss | Epoch | Step | Validation Loss |
51
  |:-------------:|:------:|:----:|:---------------:|
52
- | 13.9396 | 1.0 | 933 | 1.6778 |
53
- | 13.2393 | 2.0 | 1866 | 1.6715 |
54
- | 13.1078 | 3.0 | 2799 | 1.6800 |
55
- | 13.0497 | 4.0 | 3732 | 1.6581 |
56
- | 13.0853 | 4.9949 | 4660 | 1.6590 |
57
 
58
 
59
  ### Framework versions
 
16
 
17
  This model is a fine-tuned version of [meta-llama/Llama-3.2-1B-Instruct](https://huggingface.co/meta-llama/Llama-3.2-1B-Instruct) on an unknown dataset.
18
  It achieves the following results on the evaluation set:
19
+ - Loss: 1.6620
20
 
21
  ## Model description
22
 
 
35
  ### Training hyperparameters
36
 
37
  The following hyperparameters were used during training:
38
+ - learning_rate: 6e-05
39
  - train_batch_size: 1
40
  - eval_batch_size: 1
41
  - seed: 42
 
43
  - total_train_batch_size: 8
44
  - optimizer: Use OptimizerNames.ADAMW_TORCH 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: 1000
47
  - num_epochs: 5
48
 
49
  ### Training results
50
 
51
  | Training Loss | Epoch | Step | Validation Loss |
52
  |:-------------:|:------:|:----:|:---------------:|
53
+ | 21.2622 | 1.0 | 933 | 1.7363 |
54
+ | 13.5357 | 2.0 | 1866 | 1.6952 |
55
+ | 13.1942 | 3.0 | 2799 | 1.6826 |
56
+ | 13.0838 | 4.0 | 3732 | 1.6599 |
57
+ | 13.0959 | 4.9949 | 4660 | 1.6620 |
58
 
59
 
60
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4d0e53cc8dd5b8ab7690592c15d39e838669a505b62164f9d96401eb493be9fb
3
  size 2471645608
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:96cb3517f98807c5b70929ed89a2f2436f1114316b05129d461b8ceb3284e595
3
  size 2471645608