End of training

README.md

@@ -15,14 +15,14 @@ This student model is distilled from the teacher model [roneneldan/TinyStories-3
 The [Distily](https://github.com/lapp0/distily) library was used for this distillation.
 
 It achieves the following results on the evaluation set:
-- eval_enwikippl: 4839.0244
-- eval_frwikippl: 39016.5469
-- eval_zhwikippl: 56057.3555
-- eval_tinystoriesppl: 1738.1987
-- eval_loss: 4.8480
-- eval_runtime: 13.0187
-- eval_samples_per_second: 76.813
-- eval_steps_per_second: 9.602
+- eval_enwikippl: 133.3661
+- eval_frwikippl: 19650.0977
+- eval_zhwikippl: 54146.3867
+- eval_tinystoriesppl: 9.1470
+- eval_loss: 1.2078
+- eval_runtime: 12.9975
+- eval_samples_per_second: 76.938
+- eval_steps_per_second: 9.617
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 should probably proofread and complete it, then remove this comment.
@@ -47,7 +47,7 @@ More information needed
 The following hyperparameters were used during training:
 - distillation_objective: DistillationObjective(logits_loss_component=LossComponent(label=logits, weight=1, loss_fn=kl, layer_mapper=None, projector=None), hs_loss_component=LossComponent(label=hs, weight=0, loss_fn=None, layer_mapper=None, projector=None), attn_loss_component=LossComponent(label=attn, weight=0, loss_fn=None, layer_mapper=None, projector=None))
 - train_embeddings: True
-- learning_rate: 1e-06
+- learning_rate: 4e-06
 - train_batch_size: 1
 - eval_batch_size: 8
 - seed: 42
@@ -56,33 +56,33 @@ The following hyperparameters were used during training:
 - num_epochs: 1.0
 
 ### Resource Usage
-Peak GPU Memory: 6.6048 GB
+Peak GPU Memory: 6.6064 GB
 
 ### Eval-Phase Metrics
 | step | epoch | enwikippl | frwikippl | loss | runtime | samples_per_second | steps_per_second | tinystoriesppl | zhwikippl |
 | --- | --- | --- | --- | --- | --- | --- | --- | --- | --- |
 | **teacher eval** |  | 169.9865 | 47377.9414 |  |  |  |  | 3.9789 | 4998.1294 |
-| 0 | 0 | 23326.0117 | 54995.6562 | 6.1135 | 13.0206 | 76.802 | 9.6 | 17003.8281 | 65438.9492 |
-| 5000 | 0.0505 | 4839.0244 | 39016.5469 | 4.8478 | 13.0137 | 76.842 | 9.605 | 1738.1987 | 56102.2266 |
-| 10000 | 0.1010 | 4839.0244 | 39016.5469 | 4.8480 | 12.9688 | 77.108 | 9.639 | 1738.1987 | 56087.25 |
-| 15000 | 0.1515 | 4839.0244 | 39016.5469 | 4.8488 | 12.9532 | 77.201 | 9.65 | 1739.0610 | 56102.2266 |
-| 20000 | 0.2020 | 4839.0244 | 39016.5469 | 4.8488 | 13.014 | 76.84 | 9.605 | 1738.4856 | 56072.2734 |
-| 25000 | 0.2525 | 4839.0244 | 39016.5469 | 4.8485 | 13.0041 | 76.899 | 9.612 | 1738.1987 | 56057.3555 |
-| 30000 | 0.3030 | 4839.0244 | 39016.5469 | 4.8488 | 12.9823 | 77.028 | 9.629 | 1738.4856 | 56057.3555 |
-| 35000 | 0.3535 | 4839.0244 | 39016.5469 | 4.8485 | 13.037 | 76.705 | 9.588 | 1739.0610 | 56072.2734 |
-| 40000 | 0.4040 | 4839.0244 | 39016.5469 | 4.8482 | 12.9918 | 76.972 | 9.621 | 1738.1987 | 56057.3555 |
-| 45000 | 0.4545 | 4839.0244 | 38994.5625 | 4.8485 | 12.9951 | 76.952 | 9.619 | 1738.7732 | 56057.3555 |
-| 50000 | 0.5051 | 4839.0244 | 39016.5469 | 4.8482 | 12.9731 | 77.083 | 9.635 | 1739.3488 | 56072.2734 |
-| 55000 | 0.5556 | 4839.0244 | 39016.5469 | 4.8478 | 12.9694 | 77.105 | 9.638 | 1738.4856 | 56057.3555 |
-| 60000 | 0.6061 | 4839.0244 | 39016.5469 | 4.8488 | 13.0353 | 76.715 | 9.589 | 1738.7732 | 56057.3555 |
-| 65000 | 0.6566 | 4839.0244 | 39016.5469 | 4.8478 | 13.0087 | 76.872 | 9.609 | 1738.1987 | 56057.3555 |
-| 70000 | 0.7071 | 4839.0244 | 39016.5469 | 4.8485 | 13.033 | 76.728 | 9.591 | 1738.1987 | 56057.3555 |
-| 75000 | 0.7576 | 4839.0244 | 39016.5469 | 4.8480 | 13.0328 | 76.729 | 9.591 | 1738.4856 | 56057.3555 |
-| 80000 | 0.8081 | 4839.0244 | 39016.5469 | 4.8482 | 12.9884 | 76.992 | 9.624 | 1737.9111 | 56057.3555 |
-| 85000 | 0.8586 | 4839.0244 | 39016.5469 | 4.8485 | 13.0047 | 76.895 | 9.612 | 1738.1987 | 56087.25 |
-| 90000 | 0.9091 | 4839.0244 | 39016.5469 | 4.8478 | 13.0255 | 76.772 | 9.597 | 1738.1987 | 56057.3555 |
-| 95000 | 0.9596 | 4839.0244 | 39016.5469 | 4.8482 | 13.0007 | 76.919 | 9.615 | 1738.1987 | 56057.3555 |
-| 99000 | 1.0 | 4839.0244 | 39016.5469 | 4.8480 | 13.0187 | 76.813 | 9.602 | 1738.1987 | 56057.3555 |
+| 0 | 0 | 50480.5703 | 85684.4844 | 6.8305 | 13.0278 | 76.759 | 9.595 | 33932.0586 | 94692.1562 |
+| 5000 | 0.0505 | 133.2834 | 19661.1758 | 1.2085 | 13.0193 | 76.809 | 9.601 | 9.1500 | 54349.0273 |
+| 10000 | 0.1010 | 133.3712 | 19627.9590 | 1.2079 | 13.0358 | 76.712 | 9.589 | 9.1530 | 54117.5273 |
+| 15000 | 0.1515 | 133.2834 | 19650.0977 | 1.2077 | 13.0514 | 76.62 | 9.578 | 9.1402 | 54088.6328 |
+| 20000 | 0.2020 | 133.3299 | 19639.0254 | 1.2077 | 13.0057 | 76.889 | 9.611 | 9.1545 | 54204.1992 |
+| 25000 | 0.2525 | 133.4539 | 19650.0977 | 1.2079 | 13.0301 | 76.745 | 9.593 | 9.1538 | 54349.0273 |
+| 30000 | 0.3030 | 133.5160 | 19650.0977 | 1.2079 | 13.03 | 76.746 | 9.593 | 9.1542 | 54088.6328 |
+| 35000 | 0.3535 | 133.2834 | 19627.9590 | 1.2078 | 13.0569 | 76.588 | 9.573 | 9.1451 | 54117.5273 |
+| 40000 | 0.4040 | 133.3712 | 19627.9590 | 1.2078 | 12.9991 | 76.928 | 9.616 | 9.1523 | 54146.3867 |
+| 45000 | 0.4545 | 133.3041 | 19650.0977 | 1.2077 | 12.9923 | 76.969 | 9.621 | 9.1477 | 54088.6328 |
+| 50000 | 0.5051 | 133.2834 | 19650.0977 | 1.2078 | 13.1989 | 75.764 | 9.47 | 9.1470 | 54204.1992 |
+| 55000 | 0.5556 | 133.4953 | 19650.0977 | 1.2078 | 13.1556 | 76.013 | 9.502 | 9.1485 | 54117.5273 |
+| 60000 | 0.6061 | 133.4901 | 19661.1758 | 1.2077 | 13.206 | 75.723 | 9.465 | 9.1477 | 54117.5273 |
+| 65000 | 0.6566 | 133.4488 | 19650.0977 | 1.2077 | 13.0052 | 76.892 | 9.612 | 9.1470 | 54117.5273 |
+| 70000 | 0.7071 | 133.3661 | 19650.0977 | 1.2078 | 12.9996 | 76.925 | 9.616 | 9.1470 | 54117.5273 |
+| 75000 | 0.7576 | 133.4074 | 19650.0977 | 1.2079 | 13.0082 | 76.874 | 9.609 | 9.1470 | 54117.5273 |
+| 80000 | 0.8081 | 133.4488 | 19650.0977 | 1.2078 | 12.9816 | 77.032 | 9.629 | 9.1485 | 54117.5273 |
+| 85000 | 0.8586 | 133.3661 | 19650.0977 | 1.2077 | 12.9875 | 76.997 | 9.625 | 9.1470 | 54117.5273 |
+| 90000 | 0.9091 | 133.3661 | 19650.0977 | 1.2077 | 12.985 | 77.012 | 9.626 | 9.1462 | 54117.5273 |
+| 95000 | 0.9596 | 133.4074 | 19650.0977 | 1.2078 | 13.0478 | 76.641 | 9.58 | 9.1470 | 54146.3867 |
+| 99000 | 1.0 | 133.3661 | 19650.0977 | 1.2078 | 12.9975 | 76.938 | 9.617 | 9.1470 | 54146.3867 |
 
 ### Framework versions
 - Distily 0.2.0

logs/copy_teacher_modules=_(_lm_head___False)_, learning_rate=4e-06/events.out.tfevents.1724012794.5f530b1cf724

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a83ba5dba130affcfe3553bfe73b6e9eeadd76f9c351871e8c5c0a98ee6166bb
+size 312