yzhuang commited on
Commit
1135ae7
1 Parent(s): e1e97bd

End of training

Browse files
Files changed (2) hide show
  1. README.md +4 -4
  2. generation_config.json +3 -0
README.md CHANGED
@@ -1,6 +1,6 @@
1
  ---
2
- license: mit
3
- base_model: microsoft/phi-1_5
4
  tags:
5
  - trl
6
  - sft
@@ -17,7 +17,7 @@ should probably proofread and complete it, then remove this comment. -->
17
 
18
  # phi-1_5_fictional
19
 
20
- This model is a fine-tuned version of [microsoft/phi-1_5](https://huggingface.co/microsoft/phi-1_5) on the mmlu_no_train dataset.
21
 
22
  ## Model description
23
 
@@ -36,7 +36,7 @@ More information needed
36
  ### Training hyperparameters
37
 
38
  The following hyperparameters were used during training:
39
- - learning_rate: 1e-05
40
  - train_batch_size: 2
41
  - eval_batch_size: 2
42
  - seed: 42
 
1
  ---
2
+ license: other
3
+ base_model: facebook/opt-350m
4
  tags:
5
  - trl
6
  - sft
 
17
 
18
  # phi-1_5_fictional
19
 
20
+ This model is a fine-tuned version of [facebook/opt-350m](https://huggingface.co/facebook/opt-350m) on the mmlu_no_train dataset.
21
 
22
  ## Model description
23
 
 
36
  ### Training hyperparameters
37
 
38
  The following hyperparameters were used during training:
39
+ - learning_rate: 5e-05
40
  - train_batch_size: 2
41
  - eval_batch_size: 2
42
  - seed: 42
generation_config.json CHANGED
@@ -1,4 +1,7 @@
1
  {
2
  "_from_model_config": true,
 
 
 
3
  "transformers_version": "4.36.2"
4
  }
 
1
  {
2
  "_from_model_config": true,
3
+ "bos_token_id": 2,
4
+ "eos_token_id": 2,
5
+ "pad_token_id": 1,
6
  "transformers_version": "4.36.2"
7
  }