PatrickTyBrown commited on
Commit
ae64bf2
·
1 Parent(s): 56f0365

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +11 -4
README.md CHANGED
@@ -12,6 +12,13 @@ should probably proofread and complete it, then remove this comment. -->
12
  # GPT-Neo_DnD
13
 
14
  This model is a fine-tuned version of [PatrickTyBrown/GPT-Neo_DnD](https://huggingface.co/PatrickTyBrown/GPT-Neo_DnD) on the None dataset.
 
 
 
 
 
 
 
15
 
16
  ## Model description
17
 
@@ -30,13 +37,13 @@ More information needed
30
  ### Training hyperparameters
31
 
32
  The following hyperparameters were used during training:
33
- - learning_rate: 1.5e-05
34
- - train_batch_size: 8
35
- - eval_batch_size: 8
36
  - seed: 42
37
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
38
  - lr_scheduler_type: linear
39
- - num_epochs: 1
40
 
41
  ### Framework versions
42
 
 
12
  # GPT-Neo_DnD
13
 
14
  This model is a fine-tuned version of [PatrickTyBrown/GPT-Neo_DnD](https://huggingface.co/PatrickTyBrown/GPT-Neo_DnD) on the None dataset.
15
+ It achieves the following results on the evaluation set:
16
+ - eval_loss: 3.6958
17
+ - eval_runtime: 88.8089
18
+ - eval_samples_per_second: 6.508
19
+ - eval_steps_per_second: 3.254
20
+ - epoch: 0.93
21
+ - step: 5400
22
 
23
  ## Model description
24
 
 
37
  ### Training hyperparameters
38
 
39
  The following hyperparameters were used during training:
40
+ - learning_rate: 2.5e-05
41
+ - train_batch_size: 2
42
+ - eval_batch_size: 2
43
  - seed: 42
44
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
45
  - lr_scheduler_type: linear
46
+ - num_epochs: 3
47
 
48
  ### Framework versions
49