Hugging Face's logo

ChayanM
/
ViT-RadBert_Mimic

Image-to-Text
Transformers
Safetensors
vision-encoder-decoder
Generated from Trainer
Model card Files
xet
Community
ChayanM commited on
Commit
9e5baef
·
verified ·
1 Parent(s): ffeb7f6

Model save

Browse files
Files changed (2) hide show
  1. README.md +12 -14
  2. model.safetensors +1 -1
README.md CHANGED
@@ -15,12 +15,12 @@ should probably proofread and complete it, then remove this comment. -->
15
 
16
  This model is a fine-tuned version of [](https://huggingface.co/) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
- - Loss: 0.2458
19
- - Rouge1: 27.3521
20
- - Rouge2: 19.396
21
- - Rougel: 27.216
22
- - Rougelsum: 27.4189
23
- - Gen Len: 19.57
24
 
25
  ## Model description
26
 
@@ -45,17 +45,15 @@ The following hyperparameters were used during training:
45
  - seed: 42
46
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
47
  - lr_scheduler_type: linear
48
- - num_epochs: 5
49
 
50
  ### Training results
51
 
52
- | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
53
- |:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|:-------:|:---------:|:-------:|
54
- | 0.2329 | 1.0 | 1000 | 0.2703 | 14.7579 | 6.7085 | 14.7153 | 14.7411 | 20.0 |
55
- | 0.2345 | 2.0 | 2000 | 0.2495 | 25.2208 | 16.6545 | 25.1321 | 25.2044 | 20.0 |
56
- | 0.2211 | 3.0 | 3000 | 0.2476 | 22.7652 | 11.461 | 22.6814 | 22.7536 | 19.822 |
57
- | 0.2266 | 4.0 | 4000 | 0.2449 | 27.2342 | 17.4874 | 26.9883 | 27.1183 | 19.782 |
58
- | 0.2033 | 5.0 | 5000 | 0.2458 | 27.3521 | 19.396 | 27.216 | 27.4189 | 19.57 |
59
 
60
 
61
  ### Framework versions
 
15
 
16
  This model is a fine-tuned version of [](https://huggingface.co/) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
+ - Loss: 0.6978
19
+ - Rouge1: 0.0
20
+ - Rouge2: 0.0
21
+ - Rougel: 0.0
22
+ - Rougelsum: 0.0
23
+ - Gen Len: 20.0
24
 
25
  ## Model description
26
 
 
45
  - seed: 42
46
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
47
  - lr_scheduler_type: linear
48
+ - num_epochs: 3
49
 
50
  ### Training results
51
 
52
+ | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
53
+ |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:|:-------:|
54
+ | 0.2492 | 1.0 | 2500 | 0.5574 | 0.0 | 0.0 | 0.0 | 0.0 | 20.0 |
55
+ | 0.2334 | 2.0 | 5000 | 0.7117 | 0.0 | 0.0 | 0.0 | 0.0 | 20.0 |
56
+ | 0.2281 | 3.0 | 7500 | 0.6978 | 0.0 | 0.0 | 0.0 | 0.0 | 20.0 |
 
 
57
 
58
 
59
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1b9971c047b8ded2903e69a657adb3a5f167bc15e1fc4f172eb73a32a12e8cd5
3
  size 1834458276
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6f3ff59b022379517352b6055a5cc3ece6819ddcaa3b7d4ad724d8b009b1e8d6
3
  size 1834458276