|
2021-05-27 21:46:40,498 INFO __main__ Namespace(adjust_lr=False, config='torchdistill/configs/sample/glue/sst2/ce/bert_base_uncased.yaml', log='log/glue/sst2/ce/bert_base_uncased.txt', private_output='leaderboard/glue/standard/bert_base_uncased/', seed=None, student_only=False, task_name='sst2', test_only=False, world_size=1) |
|
2021-05-27 21:46:40,527 INFO __main__ Distributed environment: NO |
|
Num processes: 1 |
|
Process index: 0 |
|
Local process index: 0 |
|
Device: cuda |
|
Use FP16 precision: True |
|
|
|
2021-05-27 21:46:45,657 WARNING datasets.builder Reusing dataset glue (/root/.cache/huggingface/datasets/glue/sst2/1.0.0/dacbe3125aa31d7f70367a07a8a9e72a5a0bfeb5fc42e75c9db75b96da6053ad) |
|
2021-05-27 21:46:49,521 INFO __main__ Start training |
|
2021-05-27 21:46:49,521 INFO torchdistill.models.util [student model] |
|
2021-05-27 21:46:49,521 INFO torchdistill.models.util Using the original student model |
|
2021-05-27 21:46:49,522 INFO torchdistill.core.training Loss = 1.0 * OrgLoss |
|
2021-05-27 21:46:52,024 INFO torchdistill.misc.log Epoch: [0] [ 0/4210] eta: 0:08:32 lr: 2.9997624703087888e-05 sample/s: 34.45268233380702 loss: 0.6865 (0.6865) time: 0.1218 data: 0.0057 max mem: 1860 |
|
2021-05-27 21:47:42,462 INFO torchdistill.misc.log Epoch: [0] [ 500/4210] eta: 0:06:14 lr: 2.880997624703088e-05 sample/s: 49.37030933659777 loss: 0.2803 (0.3598) time: 0.0981 data: 0.0017 max mem: 2451 |
|
2021-05-27 21:48:32,727 INFO torchdistill.misc.log Epoch: [0] [1000/4210] eta: 0:05:23 lr: 2.7622327790973873e-05 sample/s: 39.871515416533974 loss: 0.1317 (0.3051) time: 0.0976 data: 0.0017 max mem: 2451 |
|
2021-05-27 21:49:23,257 INFO torchdistill.misc.log Epoch: [0] [1500/4210] eta: 0:04:33 lr: 2.6434679334916863e-05 sample/s: 35.84561181605885 loss: 0.1454 (0.2805) time: 0.0978 data: 0.0016 max mem: 2519 |
|
2021-05-27 21:50:13,560 INFO torchdistill.misc.log Epoch: [0] [2000/4210] eta: 0:03:42 lr: 2.5247030878859857e-05 sample/s: 44.91050137859036 loss: 0.0586 (0.2644) time: 0.0993 data: 0.0017 max mem: 2519 |
|
2021-05-27 21:51:03,797 INFO torchdistill.misc.log Epoch: [0] [2500/4210] eta: 0:02:52 lr: 2.405938242280285e-05 sample/s: 39.80218023083402 loss: 0.1723 (0.2589) time: 0.1047 data: 0.0017 max mem: 2524 |
|
2021-05-27 21:51:54,269 INFO torchdistill.misc.log Epoch: [0] [3000/4210] eta: 0:02:01 lr: 2.2871733966745845e-05 sample/s: 44.8911543329293 loss: 0.0482 (0.2504) time: 0.0993 data: 0.0016 max mem: 2524 |
|
2021-05-27 21:52:44,685 INFO torchdistill.misc.log Epoch: [0] [3500/4210] eta: 0:01:11 lr: 2.1684085510688836e-05 sample/s: 44.72767310855592 loss: 0.0169 (0.2495) time: 0.0991 data: 0.0017 max mem: 2524 |
|
2021-05-27 21:53:34,615 INFO torchdistill.misc.log Epoch: [0] [4000/4210] eta: 0:00:21 lr: 2.049643705463183e-05 sample/s: 44.74425402311726 loss: 0.0601 (0.2467) time: 0.0963 data: 0.0017 max mem: 2524 |
|
2021-05-27 21:53:55,646 INFO torchdistill.misc.log Epoch: [0] Total time: 0:07:03 |
|
2021-05-27 21:53:57,373 INFO /usr/local/lib/python3.7/dist-packages/datasets/metric.py Removing /root/.cache/huggingface/metrics/glue/sst2/default_experiment-1-0.arrow |
|
2021-05-27 21:53:57,374 INFO __main__ Validation: accuracy = 0.9197247706422018 |
|
2021-05-27 21:53:57,374 INFO __main__ Updating ckpt at ./resource/ckpt/glue/sst2/ce/sst2-bert-base-uncased |
|
2021-05-27 21:53:58,421 INFO torchdistill.misc.log Epoch: [1] [ 0/4210] eta: 0:08:17 lr: 1.9997624703087886e-05 sample/s: 35.521468859567406 loss: 0.0474 (0.0474) time: 0.1181 data: 0.0055 max mem: 2524 |
|
2021-05-27 21:54:48,646 INFO torchdistill.misc.log Epoch: [1] [ 500/4210] eta: 0:06:12 lr: 1.880997624703088e-05 sample/s: 36.44645848042144 loss: 0.0000 (0.2791) time: 0.1028 data: 0.0017 max mem: 2524 |
|
2021-05-27 21:55:38,247 INFO torchdistill.misc.log Epoch: [1] [1000/4210] eta: 0:05:20 lr: 1.7622327790973874e-05 sample/s: 35.710640471680044 loss: 0.0000 (0.3181) time: 0.1011 data: 0.0018 max mem: 2524 |
|
2021-05-27 21:56:28,855 INFO torchdistill.misc.log Epoch: [1] [1500/4210] eta: 0:04:31 lr: 1.6434679334916864e-05 sample/s: 39.87085213729446 loss: 0.0165 (0.3264) time: 0.0996 data: 0.0017 max mem: 2524 |
|
2021-05-27 21:57:18,830 INFO torchdistill.misc.log Epoch: [1] [2000/4210] eta: 0:03:41 lr: 1.5247030878859858e-05 sample/s: 45.886779241949334 loss: 0.0044 (0.3258) time: 0.0958 data: 0.0017 max mem: 2524 |
|
2021-05-27 21:58:09,046 INFO torchdistill.misc.log Epoch: [1] [2500/4210] eta: 0:02:51 lr: 1.405938242280285e-05 sample/s: 39.92987550188378 loss: 0.1085 (0.3327) time: 0.1015 data: 0.0017 max mem: 2524 |
|
2021-05-27 21:58:58,696 INFO torchdistill.misc.log Epoch: [1] [3000/4210] eta: 0:02:01 lr: 1.2871733966745844e-05 sample/s: 45.834253539904 loss: 0.0002 (0.3333) time: 0.0970 data: 0.0017 max mem: 2524 |
|
2021-05-27 21:59:48,373 INFO torchdistill.misc.log Epoch: [1] [3500/4210] eta: 0:01:10 lr: 1.1684085510688835e-05 sample/s: 44.44754822352508 loss: 0.5874 (0.3352) time: 0.0967 data: 0.0017 max mem: 2524 |
|
2021-05-27 22:00:38,264 INFO torchdistill.misc.log Epoch: [1] [4000/4210] eta: 0:00:20 lr: 1.0496437054631829e-05 sample/s: 39.99126625063763 loss: 0.0001 (0.3336) time: 0.1006 data: 0.0017 max mem: 2524 |
|
2021-05-27 22:00:59,160 INFO torchdistill.misc.log Epoch: [1] Total time: 0:07:00 |
|
2021-05-27 22:01:00,882 INFO /usr/local/lib/python3.7/dist-packages/datasets/metric.py Removing /root/.cache/huggingface/metrics/glue/sst2/default_experiment-1-0.arrow |
|
2021-05-27 22:01:00,882 INFO __main__ Validation: accuracy = 0.9185779816513762 |
|
2021-05-27 22:01:00,987 INFO torchdistill.misc.log Epoch: [2] [ 0/4210] eta: 0:07:20 lr: 9.997624703087887e-06 sample/s: 39.848786999254195 loss: 0.6077 (0.6077) time: 0.1046 data: 0.0042 max mem: 2524 |
|
2021-05-27 22:01:50,624 INFO torchdistill.misc.log Epoch: [2] [ 500/4210] eta: 0:06:08 lr: 8.809976247030879e-06 sample/s: 36.41133843639655 loss: 0.0000 (0.1890) time: 0.0982 data: 0.0018 max mem: 2524 |
|
2021-05-27 22:02:40,203 INFO torchdistill.misc.log Epoch: [2] [1000/4210] eta: 0:05:18 lr: 7.622327790973873e-06 sample/s: 49.212313919892054 loss: 0.0000 (0.1807) time: 0.1030 data: 0.0017 max mem: 2524 |
|
2021-05-27 22:03:30,223 INFO torchdistill.misc.log Epoch: [2] [1500/4210] eta: 0:04:29 lr: 6.434679334916865e-06 sample/s: 40.599501012736035 loss: 0.0000 (0.1792) time: 0.1018 data: 0.0017 max mem: 2524 |
|
2021-05-27 22:04:19,806 INFO torchdistill.misc.log Epoch: [2] [2000/4210] eta: 0:03:39 lr: 5.247030878859857e-06 sample/s: 50.41261065270825 loss: 0.0000 (0.1790) time: 0.0993 data: 0.0018 max mem: 2524 |
|
2021-05-27 22:05:09,793 INFO torchdistill.misc.log Epoch: [2] [2500/4210] eta: 0:02:50 lr: 4.0593824228028505e-06 sample/s: 33.1207489137237 loss: 0.0000 (0.1839) time: 0.1056 data: 0.0017 max mem: 2524 |
|
2021-05-27 22:05:59,813 INFO torchdistill.misc.log Epoch: [2] [3000/4210] eta: 0:02:00 lr: 2.8717339667458436e-06 sample/s: 40.70687400369286 loss: 0.0000 (0.1804) time: 0.1029 data: 0.0017 max mem: 2524 |
|
2021-05-27 22:06:49,594 INFO torchdistill.misc.log Epoch: [2] [3500/4210] eta: 0:01:10 lr: 1.684085510688836e-06 sample/s: 35.86921598433294 loss: 0.0000 (0.1842) time: 0.1023 data: 0.0017 max mem: 2524 |
|
2021-05-27 22:07:39,232 INFO torchdistill.misc.log Epoch: [2] [4000/4210] eta: 0:00:20 lr: 4.96437054631829e-07 sample/s: 39.971353080901814 loss: 0.0000 (0.1840) time: 0.0981 data: 0.0018 max mem: 2524 |
|
2021-05-27 22:07:59,740 INFO torchdistill.misc.log Epoch: [2] Total time: 0:06:58 |
|
2021-05-27 22:08:01,462 INFO /usr/local/lib/python3.7/dist-packages/datasets/metric.py Removing /root/.cache/huggingface/metrics/glue/sst2/default_experiment-1-0.arrow |
|
2021-05-27 22:08:01,462 INFO __main__ Validation: accuracy = 0.9254587155963303 |
|
2021-05-27 22:08:01,462 INFO __main__ Updating ckpt at ./resource/ckpt/glue/sst2/ce/sst2-bert-base-uncased |
|
2021-05-27 22:08:06,304 INFO __main__ [Student: bert-base-uncased] |
|
2021-05-27 22:08:08,039 INFO /usr/local/lib/python3.7/dist-packages/datasets/metric.py Removing /root/.cache/huggingface/metrics/glue/sst2/default_experiment-1-0.arrow |
|
2021-05-27 22:08:08,039 INFO __main__ Test: accuracy = 0.9254587155963303 |
|
2021-05-27 22:08:08,040 INFO __main__ Start prediction for private dataset(s) |
|
2021-05-27 22:08:08,040 INFO __main__ sst2/test: 1821 samples |
|
|
