SashaSheykina commited on
Commit
3e2d185
1 Parent(s): 7b1a076

Training complete

Browse files
README.md CHANGED
@@ -19,13 +19,13 @@ should probably proofread and complete it, then remove this comment. -->
19
 
20
  This model is a fine-tuned version of [Salesforce/codet5-base](https://huggingface.co/Salesforce/codet5-base) on an unknown dataset.
21
  It achieves the following results on the evaluation set:
22
- - Loss: 0.4358
23
- - Rouge1: 0.3457
24
- - Rouge2: 0.1202
25
- - Rougel: 0.3148
26
- - Bleu: 12.8425
27
- - Meteor: 0.3899
28
- - Codebleu: 0.1722
29
 
30
  ## Model description
31
 
@@ -45,11 +45,11 @@ More information needed
45
 
46
  The following hyperparameters were used during training:
47
  - learning_rate: 5e-05
48
- - train_batch_size: 6
49
- - eval_batch_size: 12
50
  - seed: 42
51
- - gradient_accumulation_steps: 2
52
- - total_train_batch_size: 12
53
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
54
  - lr_scheduler_type: linear
55
  - num_epochs: 3
 
19
 
20
  This model is a fine-tuned version of [Salesforce/codet5-base](https://huggingface.co/Salesforce/codet5-base) on an unknown dataset.
21
  It achieves the following results on the evaluation set:
22
+ - Loss: 0.7559
23
+ - Rouge1: 0.1769
24
+ - Rouge2: 0.0365
25
+ - Rougel: 0.1659
26
+ - Bleu: 3.0318
27
+ - Meteor: 0.2731
28
+ - Codebleu: 0.0562
29
 
30
  ## Model description
31
 
 
45
 
46
  The following hyperparameters were used during training:
47
  - learning_rate: 5e-05
48
+ - train_batch_size: 2
49
+ - eval_batch_size: 2
50
  - seed: 42
51
+ - gradient_accumulation_steps: 16
52
+ - total_train_batch_size: 32
53
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
54
  - lr_scheduler_type: linear
55
  - num_epochs: 3
runs/Aug05_10-21-39_cb69df8c9ea7/events.out.tfevents.1722854125.cb69df8c9ea7.4878.8 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d0791c1a84c57c1da8cc484fbd69ed4c841e1eadaa67136880084d3ca9d0afdc
3
+ size 648