| 2021-05-21 21:05:54,150 INFO __main__ Namespace(adjust_lr=False, config='torchdistill/configs/sample/glue/stsb/mse/bert_large_uncased.yaml', log='log/glue/stsb/mse/bert_large_uncased.txt', private_output='leaderboard/glue/standard/bert_large_uncased/', seed=None, student_only=False, task_name='stsb', test_only=False, world_size=1) | |
| 2021-05-21 21:05:54,184 INFO __main__ Distributed environment: NO | |
| Num processes: 1 | |
| Process index: 0 | |
| Local process index: 0 | |
| Device: cuda | |
| Use FP16 precision: True | |
| 2021-05-21 21:06:08,347 INFO __main__ Start training | |
| 2021-05-21 21:06:08,348 INFO torchdistill.models.util [student model] | |
| 2021-05-21 21:06:08,348 INFO torchdistill.models.util Using the original student model | |
| 2021-05-21 21:06:08,348 INFO torchdistill.core.training Loss = 1.0 * OrgLoss | |
| 2021-05-21 21:06:11,614 INFO torchdistill.misc.log Epoch: [0] [ 0/180] eta: 0:01:58 lr: 1.9962962962962963e-05 sample/s: 6.150729612978653 loss: 11.0822 (11.0822) time: 0.6556 data: 0.0052 max mem: 4716 | |
| 2021-05-21 21:06:50,634 INFO torchdistill.misc.log Epoch: [0] [ 50/180] eta: 0:01:41 lr: 1.8111111111111112e-05 sample/s: 4.650817812095187 loss: 5.7362 (7.8091) time: 0.8036 data: 0.0036 max mem: 10866 | |
| 2021-05-21 21:07:31,077 INFO torchdistill.misc.log Epoch: [0] [100/180] eta: 0:01:03 lr: 1.625925925925926e-05 sample/s: 5.073427520805109 loss: 1.5491 (5.1515) time: 0.7817 data: 0.0037 max mem: 10866 | |
| 2021-05-21 21:08:12,096 INFO torchdistill.misc.log Epoch: [0] [150/180] eta: 0:00:24 lr: 1.4407407407407407e-05 sample/s: 4.66012418875246 loss: 0.6607 (3.6948) time: 0.8202 data: 0.0037 max mem: 12375 | |
| 2021-05-21 21:08:33,652 INFO torchdistill.misc.log Epoch: [0] Total time: 0:02:22 | |
| 2021-05-21 21:08:42,643 INFO /usr/local/lib/python3.7/dist-packages/datasets/metric.py Removing /root/.cache/huggingface/metrics/glue/stsb/default_experiment-1-0.arrow | |
| 2021-05-21 21:08:42,644 INFO __main__ Validation: pearson = 0.869070919965039, spearmanr = 0.8715366411318595 | |
| 2021-05-21 21:08:42,644 INFO __main__ Updating ckpt | |
| 2021-05-21 21:08:48,054 INFO torchdistill.misc.log Epoch: [1] [ 0/180] eta: 0:02:38 lr: 1.3296296296296298e-05 sample/s: 4.560549161662958 loss: 0.5407 (0.5407) time: 0.8833 data: 0.0062 max mem: 12375 | |
| 2021-05-21 21:09:27,646 INFO torchdistill.misc.log Epoch: [1] [ 50/180] eta: 0:01:43 lr: 1.1444444444444444e-05 sample/s: 5.076204414368103 loss: 0.3929 (0.4579) time: 0.7556 data: 0.0038 max mem: 12375 | |
| 2021-05-21 21:10:06,919 INFO torchdistill.misc.log Epoch: [1] [100/180] eta: 0:01:03 lr: 9.592592592592593e-06 sample/s: 5.071968912559318 loss: 0.3982 (0.4221) time: 0.8143 data: 0.0039 max mem: 12375 | |
| 2021-05-21 21:10:47,745 INFO torchdistill.misc.log Epoch: [1] [150/180] eta: 0:00:23 lr: 7.74074074074074e-06 sample/s: 5.072432017980795 loss: 0.3499 (0.4130) time: 0.8114 data: 0.0038 max mem: 12375 | |
| 2021-05-21 21:11:09,859 INFO torchdistill.misc.log Epoch: [1] Total time: 0:02:22 | |
| 2021-05-21 21:11:18,716 INFO /usr/local/lib/python3.7/dist-packages/datasets/metric.py Removing /root/.cache/huggingface/metrics/glue/stsb/default_experiment-1-0.arrow | |
| 2021-05-21 21:11:18,716 INFO __main__ Validation: pearson = 0.8958498997945837, spearmanr = 0.8947816880373578 | |
| 2021-05-21 21:11:18,716 INFO __main__ Updating ckpt | |
| 2021-05-21 21:11:24,308 INFO torchdistill.misc.log Epoch: [2] [ 0/180] eta: 0:02:26 lr: 6.62962962962963e-06 sample/s: 4.964641330957722 loss: 0.4758 (0.4758) time: 0.8114 data: 0.0057 max mem: 12375 | |
| 2021-05-21 21:12:04,343 INFO torchdistill.misc.log Epoch: [2] [ 50/180] eta: 0:01:44 lr: 4.777777777777778e-06 sample/s: 5.660394215048351 loss: 0.2219 (0.2284) time: 0.7505 data: 0.0037 max mem: 12375 | |
| 2021-05-21 21:12:44,799 INFO torchdistill.misc.log Epoch: [2] [100/180] eta: 0:01:04 lr: 2.9259259259259257e-06 sample/s: 5.058339142243115 loss: 0.1955 (0.2248) time: 0.7775 data: 0.0037 max mem: 12375 | |
| 2021-05-21 21:13:24,530 INFO torchdistill.misc.log Epoch: [2] [150/180] eta: 0:00:24 lr: 1.074074074074074e-06 sample/s: 5.060421749670249 loss: 0.2013 (0.2159) time: 0.7998 data: 0.0037 max mem: 12375 | |
| 2021-05-21 21:13:47,122 INFO torchdistill.misc.log Epoch: [2] Total time: 0:02:23 | |
| 2021-05-21 21:13:55,981 INFO /usr/local/lib/python3.7/dist-packages/datasets/metric.py Removing /root/.cache/huggingface/metrics/glue/stsb/default_experiment-1-0.arrow | |
| 2021-05-21 21:13:55,982 INFO __main__ Validation: pearson = 0.8982067559574533, spearmanr = 0.8962432413562028 | |
| 2021-05-21 21:13:55,982 INFO __main__ Updating ckpt | |
| 2021-05-21 21:14:06,537 INFO __main__ [Student: bert-large-uncased] | |
| 2021-05-21 21:14:15,405 INFO /usr/local/lib/python3.7/dist-packages/datasets/metric.py Removing /root/.cache/huggingface/metrics/glue/stsb/default_experiment-1-0.arrow | |
| 2021-05-21 21:14:15,405 INFO __main__ Test: pearson = 0.8982067559574533, spearmanr = 0.8962432413562028 | |
| 2021-05-21 21:14:15,405 INFO __main__ Start prediction for private dataset(s) | |
| 2021-05-21 21:14:15,407 INFO __main__ stsb/test: 1379 samples | |