Hugging Face's logo

claudiubarbu
/
reward

Text Classification
Transformers
Safetensors
gpt2
trl
reward-trainer
Generated from Trainer
Model card Files
xet
 Community
claudiubarbu commited on
Commit
572f948
·
verified ·
1 Parent(s): 7dd5cef

HW2-reward

Browse files
Files changed (2) hide show
  1. README.md +3 -3
  2. model.safetensors +1 -1
README.md CHANGED
@@ -38,7 +38,7 @@ More information needed
38
 
39
  The following hyperparameters were used during training:
40
  - learning_rate: 5e-05
41
- - train_batch_size: 8
42
  - eval_batch_size: 8
43
  - seed: 42
44
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
@@ -52,6 +52,6 @@ The following hyperparameters were used during training:
52
  ### Framework versions
53
 
54
  - Transformers 4.44.2
55
- - Pytorch 2.4.0+cu121
56
- - Datasets 2.21.0
57
  - Tokenizers 0.19.1
 
38
 
39
  The following hyperparameters were used during training:
40
  - learning_rate: 5e-05
41
+ - train_batch_size: 4
42
  - eval_batch_size: 8
43
  - seed: 42
44
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 
52
  ### Framework versions
53
 
54
  - Transformers 4.44.2
55
+ - Pytorch 2.0.0
56
+ - Datasets 2.16.1
57
  - Tokenizers 0.19.1
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ccae9ffcd1d79f49675032dbf458d309bf7b6a6ef34d4342e447aa985dfafc03
3
  size 497780432
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5498cf68773c3edf693db3d07969c41093e036045f1102d232ad3862a6443a1b
3
  size 497780432