ppt-pythia-160m-uniform250-initial_weight_shuffle1_warmup2000-seed208-stage2
This model is a fine-tuned version of qing-yao/ppt-pythia-160m-uniform250-previous_mse-seed208-stage1 on the None dataset. It achieves the following results on the evaluation set:
- Loss: 3.8267
Model description
More information needed
Intended uses & limitations
More information needed
Training and evaluation data
More information needed
Training procedure
Training hyperparameters
The following hyperparameters were used during training:
- learning_rate: 0.001
- train_batch_size: 16
- eval_batch_size: 16
- seed: 208
- gradient_accumulation_steps: 2
- total_train_batch_size: 32
- optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
- lr_scheduler_type: cosine_with_min_lr
- lr_scheduler_warmup_steps: 2000
- training_steps: 10000
Training results
| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| 10.4934 | 0.005 | 50 | 9.6301 |
| 9.0172 | 0.01 | 100 | 8.4171 |
| 7.8323 | 0.015 | 150 | 7.4441 |
| 7.2292 | 0.02 | 200 | 7.1013 |
| 6.9124 | 0.025 | 250 | 6.8120 |
| 6.6756 | 0.03 | 300 | 6.6362 |
| 6.4899 | 0.035 | 350 | 6.4460 |
| 6.3208 | 0.04 | 400 | 6.3024 |
| 6.2162 | 0.045 | 450 | 6.1945 |
| 6.1508 | 0.05 | 500 | 6.0807 |
| 6.1066 | 0.055 | 550 | 6.0899 |
| 5.9218 | 0.06 | 600 | 5.8885 |
| 5.8216 | 0.065 | 650 | 5.8211 |
| 5.7382 | 0.07 | 700 | 5.7136 |
| 5.6549 | 0.075 | 750 | 5.6316 |
| 5.5781 | 0.08 | 800 | 5.5524 |
| 5.5053 | 0.085 | 850 | 5.4847 |
| 5.4670 | 0.09 | 900 | 5.4878 |
| 5.4370 | 0.095 | 950 | 5.3838 |
| 5.4099 | 0.1 | 1000 | 5.3349 |
| 5.3365 | 0.105 | 1050 | 5.2945 |
| 5.2360 | 0.11 | 1100 | 5.2330 |
| 5.1833 | 0.115 | 1150 | 5.1846 |
| 5.1776 | 0.12 | 1200 | 5.1540 |
| 5.1299 | 0.125 | 1250 | 5.1519 |
| 5.0981 | 0.13 | 1300 | 5.0490 |
| 5.0718 | 0.135 | 1350 | 5.0141 |
| 4.9725 | 0.14 | 1400 | 4.9371 |
| 4.9293 | 0.145 | 1450 | 4.8889 |
| 4.8702 | 0.15 | 1500 | 4.8211 |
| 4.8698 | 0.155 | 1550 | 4.7864 |
| 4.7532 | 0.16 | 1600 | 4.7486 |
| 4.7528 | 0.165 | 1650 | 4.7083 |
| 4.7059 | 0.17 | 1700 | 4.6648 |
| 4.6829 | 0.175 | 1750 | 4.6876 |
| 4.6688 | 0.18 | 1800 | 4.6451 |
| 4.6689 | 0.185 | 1850 | 5.4196 |
| 6.3669 | 0.19 | 1900 | 5.4850 |
| 5.1117 | 0.195 | 1950 | 4.8476 |
| 4.7666 | 0.2 | 2000 | 4.7002 |
| 4.6734 | 0.205 | 2050 | 4.6176 |
| 4.6132 | 0.21 | 2100 | 4.5561 |
| 4.5705 | 0.215 | 2150 | 4.5292 |
| 4.5449 | 0.22 | 2200 | 4.5178 |
| 4.5158 | 0.225 | 2250 | 4.4808 |
| 4.4901 | 0.23 | 2300 | 4.4412 |
| 4.4737 | 0.235 | 2350 | 4.4250 |
| 4.4662 | 0.24 | 2400 | 4.4152 |
| 4.4208 | 0.245 | 2450 | 4.4022 |
| 4.8741 | 0.25 | 2500 | 4.7169 |
| 4.5914 | 0.255 | 2550 | 4.4792 |
| 4.4503 | 0.26 | 2600 | 4.4214 |
| 4.4241 | 0.265 | 2650 | 4.3878 |
| 4.3963 | 0.27 | 2700 | 4.3618 |
| 4.3725 | 0.275 | 2750 | 4.3434 |
| 4.3632 | 0.28 | 2800 | 4.3409 |
| 4.3437 | 0.285 | 2850 | 4.3287 |
| 4.3340 | 0.29 | 2900 | 4.3134 |
| 4.2883 | 0.295 | 2950 | 4.2944 |
| 4.3273 | 0.3 | 3000 | 4.2779 |
| 4.2884 | 0.305 | 3050 | 4.3026 |
| 4.3034 | 0.31 | 3100 | 4.2917 |
| 4.3022 | 0.315 | 3150 | 4.2501 |
| 4.2649 | 0.32 | 3200 | 4.2196 |
| 4.2466 | 0.325 | 3250 | 4.2438 |
| 4.2694 | 0.33 | 3300 | 4.2107 |
| 4.2330 | 0.335 | 3350 | 4.2044 |
| 4.2479 | 0.34 | 3400 | 4.1814 |
| 4.2237 | 0.345 | 3450 | 4.1679 |
| 4.2051 | 0.35 | 3500 | 4.1583 |
| 4.1936 | 0.355 | 3550 | 4.1708 |
| 4.1939 | 0.36 | 3600 | 4.1811 |
| 4.2278 | 0.365 | 3650 | 4.2006 |
| 4.1940 | 0.37 | 3700 | 4.1548 |
| 4.2540 | 0.375 | 3750 | 4.1671 |
| 4.2727 | 0.38 | 3800 | 4.1950 |
| 4.1947 | 0.385 | 3850 | 4.1860 |
| 4.1820 | 0.39 | 3900 | 4.1495 |
| 4.2499 | 0.395 | 3950 | 4.1677 |
| 4.1566 | 0.4 | 4000 | 4.1319 |
| 4.1534 | 0.405 | 4050 | 4.1147 |
| 4.1498 | 0.41 | 4100 | 4.1046 |
| 4.1101 | 0.415 | 4150 | 4.0942 |
| 4.1134 | 0.42 | 4200 | 4.0981 |
| 4.1195 | 0.425 | 4250 | 4.0852 |
| 4.0960 | 0.43 | 4300 | 4.0806 |
| 4.0977 | 0.435 | 4350 | 4.0765 |
| 4.1075 | 0.44 | 4400 | 4.1195 |
| 4.1207 | 0.445 | 4450 | 4.0733 |
| 4.1031 | 0.45 | 4500 | 4.0614 |
| 4.0555 | 0.455 | 4550 | 4.0636 |
| 4.0647 | 0.46 | 4600 | 4.0523 |
| 4.0794 | 0.465 | 4650 | 4.0404 |
| 4.0778 | 0.47 | 4700 | 4.0418 |
| 4.0555 | 0.475 | 4750 | 4.0283 |
| 4.0480 | 0.48 | 4800 | 4.0224 |
| 4.0548 | 0.485 | 4850 | 4.0177 |
| 4.0499 | 0.49 | 4900 | 4.0122 |
| 4.0319 | 0.495 | 4950 | 4.0076 |
| 4.0339 | 0.5 | 5000 | 4.0175 |
| 4.0595 | 0.505 | 5050 | 4.0014 |
| 4.0450 | 0.51 | 5100 | 3.9945 |
| 4.0257 | 0.515 | 5150 | 4.0005 |
| 4.0242 | 0.52 | 5200 | 3.9899 |
| 4.0114 | 0.525 | 5250 | 3.9817 |
| 3.9979 | 0.53 | 5300 | 3.9757 |
| 3.9955 | 0.535 | 5350 | 3.9736 |
| 3.9922 | 0.54 | 5400 | 3.9722 |
| 3.9894 | 0.545 | 5450 | 3.9687 |
| 3.9989 | 0.55 | 5500 | 3.9665 |
| 3.9934 | 0.555 | 5550 | 3.9615 |
| 3.9983 | 0.56 | 5600 | 3.9621 |
| 3.9995 | 0.565 | 5650 | 3.9552 |
| 3.9800 | 0.57 | 5700 | 3.9483 |
| 3.9518 | 0.575 | 5750 | 3.9475 |
| 3.9838 | 0.58 | 5800 | 3.9436 |
| 3.9619 | 0.585 | 5850 | 3.9404 |
| 3.9762 | 0.59 | 5900 | 3.9405 |
| 3.9705 | 0.595 | 5950 | 3.9374 |
| 3.9592 | 0.6 | 6000 | 3.9267 |
| 3.9712 | 0.605 | 6050 | 3.9217 |
| 3.9679 | 0.61 | 6100 | 3.9223 |
| 3.9558 | 0.615 | 6150 | 3.9345 |
| 3.9507 | 0.62 | 6200 | 3.9307 |
| 3.9492 | 0.625 | 6250 | 3.9217 |
| 3.9658 | 0.63 | 6300 | 3.9158 |
| 3.9414 | 0.635 | 6350 | 3.9087 |
| 3.9259 | 0.64 | 6400 | 3.9095 |
| 3.9406 | 0.645 | 6450 | 3.9008 |
| 3.9161 | 0.65 | 6500 | 3.8961 |
| 3.9259 | 0.655 | 6550 | 3.8934 |
| 3.9167 | 0.66 | 6600 | 3.8910 |
| 3.9088 | 0.665 | 6650 | 3.8970 |
| 3.9062 | 0.67 | 6700 | 3.8899 |
| 3.9201 | 0.675 | 6750 | 3.8952 |
| 3.9184 | 0.68 | 6800 | 3.8867 |
| 3.9017 | 0.685 | 6850 | 3.8827 |
| 3.9037 | 0.69 | 6900 | 3.8799 |
| 3.9064 | 0.695 | 6950 | 3.8839 |
| 3.9029 | 0.7 | 7000 | 3.8842 |
| 3.8995 | 0.705 | 7050 | 3.8761 |
| 3.9481 | 0.71 | 7100 | 3.8725 |
| 3.9128 | 0.715 | 7150 | 3.8701 |
| 3.9273 | 0.72 | 7200 | 3.8666 |
| 3.9191 | 0.725 | 7250 | 3.8653 |
| 3.9198 | 0.73 | 7300 | 3.8625 |
| 3.9357 | 0.735 | 7350 | 3.8620 |
| 3.8900 | 0.74 | 7400 | 3.8601 |
| 3.8936 | 0.745 | 7450 | 3.8567 |
| 3.8950 | 0.75 | 7500 | 3.8554 |
| 3.8830 | 0.755 | 7550 | 3.8537 |
| 3.8756 | 0.76 | 7600 | 3.8526 |
| 3.8813 | 0.765 | 7650 | 3.8508 |
| 3.8874 | 0.77 | 7700 | 3.8496 |
| 3.8924 | 0.775 | 7750 | 3.8489 |
| 3.9145 | 0.78 | 7800 | 3.8518 |
| 3.8653 | 0.785 | 7850 | 3.8478 |
| 3.8984 | 0.79 | 7900 | 3.8460 |
| 3.8986 | 0.795 | 7950 | 3.8447 |
| 3.8607 | 0.8 | 8000 | 3.8468 |
| 3.8699 | 0.805 | 8050 | 3.8430 |
| 3.8648 | 0.81 | 8100 | 3.8415 |
| 3.8598 | 0.815 | 8150 | 3.8405 |
| 3.8583 | 0.82 | 8200 | 3.8400 |
| 3.8713 | 0.825 | 8250 | 3.8394 |
| 3.8775 | 0.83 | 8300 | 3.8375 |
| 3.8660 | 0.835 | 8350 | 3.8375 |
| 3.8622 | 0.84 | 8400 | 3.8383 |
| 3.8608 | 0.845 | 8450 | 3.8367 |
| 3.8698 | 0.85 | 8500 | 3.8362 |
| 3.8868 | 0.855 | 8550 | 3.8352 |
| 3.8589 | 0.86 | 8600 | 3.8347 |
| 3.8685 | 0.865 | 8650 | 3.8344 |
| 3.8558 | 0.87 | 8700 | 3.8359 |
| 3.8808 | 0.875 | 8750 | 3.8338 |
| 3.8300 | 0.88 | 8800 | 3.8362 |
| 3.8716 | 0.885 | 8850 | 3.8338 |
| 3.8759 | 0.89 | 8900 | 3.8327 |
| 3.8839 | 0.895 | 8950 | 3.8321 |
| 3.8793 | 0.9 | 9000 | 3.8314 |
| 3.8542 | 0.905 | 9050 | 3.8324 |
| 3.8520 | 0.91 | 9100 | 3.8310 |
| 3.8609 | 0.915 | 9150 | 3.8308 |
| 3.8625 | 0.92 | 9200 | 3.8303 |
| 3.8514 | 0.925 | 9250 | 3.8297 |
| 3.8676 | 0.93 | 9300 | 3.8295 |
| 3.8723 | 0.935 | 9350 | 3.8292 |
| 3.8682 | 0.94 | 9400 | 3.8288 |
| 3.8587 | 0.945 | 9450 | 3.8285 |
| 3.8627 | 0.95 | 9500 | 3.8283 |
| 3.8537 | 0.955 | 9550 | 3.8281 |
| 3.8879 | 0.96 | 9600 | 3.8277 |
| 3.8617 | 0.965 | 9650 | 3.8278 |
| 3.8567 | 0.97 | 9700 | 3.8275 |
| 3.8363 | 0.975 | 9750 | 3.8274 |
| 3.8580 | 0.98 | 9800 | 3.8276 |
| 3.8532 | 0.985 | 9850 | 3.8270 |
| 3.8987 | 0.99 | 9900 | 3.8270 |
| 3.8270 | 0.995 | 9950 | 3.8271 |
| 3.8707 | 1.0 | 10000 | 3.8267 |
Framework versions
- Transformers 5.4.0
- Pytorch 2.8.0+cu128
- Datasets 3.2.0
- Tokenizers 0.22.1