mnoukhov commited on
Commit
da2d2a3
1 Parent(s): e9a01bd

Model save

Browse files
Files changed (2) hide show
  1. README.md +8 -8
  2. model.safetensors +1 -1
README.md CHANGED
@@ -19,8 +19,8 @@ should probably proofread and complete it, then remove this comment. -->
19
 
20
  This model is a fine-tuned version of [mnoukhov/pythia160m-sft-tldr](https://huggingface.co/mnoukhov/pythia160m-sft-tldr) on an unknown dataset.
21
  It achieves the following results on the evaluation set:
22
- - Loss: 0.5486
23
- - Accuracy: 0.7121
24
 
25
  ## Model description
26
 
@@ -45,8 +45,8 @@ The following hyperparameters were used during training:
45
  - seed: 42
46
  - distributed_type: multi-GPU
47
  - num_devices: 4
48
- - gradient_accumulation_steps: 8
49
- - total_train_batch_size: 256
50
  - total_eval_batch_size: 32
51
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
52
  - lr_scheduler_type: cosine
@@ -56,10 +56,10 @@ The following hyperparameters were used during training:
56
 
57
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
58
  |:-------------:|:------:|:----:|:---------------:|:--------:|
59
- | No log | 0.2012 | 73 | 0.5667 | 0.6998 |
60
- | 0.6204 | 0.4025 | 146 | 0.5560 | 0.7120 |
61
- | 0.5593 | 0.6037 | 219 | 0.5532 | 0.7078 |
62
- | 0.5593 | 0.8050 | 292 | 0.5486 | 0.7121 |
63
 
64
 
65
  ### Framework versions
 
19
 
20
  This model is a fine-tuned version of [mnoukhov/pythia160m-sft-tldr](https://huggingface.co/mnoukhov/pythia160m-sft-tldr) on an unknown dataset.
21
  It achieves the following results on the evaluation set:
22
+ - Loss: 0.5399
23
+ - Accuracy: 0.7213
24
 
25
  ## Model description
26
 
 
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
 
56
 
57
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
58
  |:-------------:|:------:|:----:|:---------------:|:--------:|
59
+ | 0.596 | 0.2006 | 291 | 0.5819 | 0.6923 |
60
+ | 0.5382 | 0.4011 | 582 | 0.5513 | 0.7151 |
61
+ | 0.505 | 0.6017 | 873 | 0.5480 | 0.7141 |
62
+ | 0.4836 | 0.8022 | 1164 | 0.5399 | 0.7213 |
63
 
64
 
65
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6f9c43592c1a56d23791aa6c6dff2372eec63304787d3761f7be097f2f125558
3
  size 494777904
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b287adaea768b5fa7d4fdf1bb33d5fee3e53393b29a3e2868b3ebb29d710a49e
3
  size 494777904