Hugging Face's logo

Kiran2004
/
GPT2_MentalHealth_ChatBot

Text Generation
Transformers
Safetensors
English
gpt2
conversational
text-generation-inference
Model card Files
xet
 Community
Kiran2004 commited on
Commit
910f737
·
verified ·
1 Parent(s): 231505a

End of training

Browse files
Files changed (3) hide show
  1. README.md +1 -2
  2. model.safetensors +1 -1
  3. training_args.bin +1 -1
README.md CHANGED
@@ -33,7 +33,7 @@ More information needed
33
 
34
  The following hyperparameters were used during training:
35
  - learning_rate: 0.0001
36
- - train_batch_size: 2
37
  - eval_batch_size: 8
38
  - seed: 42
39
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
@@ -48,5 +48,4 @@ The following hyperparameters were used during training:
48
 
49
  - Transformers 4.38.2
50
  - Pytorch 2.2.1+cu121
51
- - Datasets 2.18.0
52
  - Tokenizers 0.15.2
 
33
 
34
  The following hyperparameters were used during training:
35
  - learning_rate: 0.0001
36
+ - train_batch_size: 3
37
  - eval_batch_size: 8
38
  - seed: 42
39
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 
48
 
49
  - Transformers 4.38.2
50
  - Pytorch 2.2.1+cu121
 
51
  - Tokenizers 0.15.2
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:40cc9ffb48b57efad5746aa85dc6a16742c8e4e3283c27c5f729eaea81b0d5ae
3
  size 497774208
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5150fa4434bace3d08d683069c882732ab94b1c015946845c92baa749818591a
3
  size 497774208
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:18e350fc74008d3190d23f83976d3226e81fd74cbdcfb902d30e9c42652910ed
3
  size 4856
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:649319d002e8c5190cf3cb47912bb940cc7bb200c91d9eeed090479529a6ba5c
3
  size 4856