2021-05-27 20:30:20,874 INFO __main__ Namespace(adjust_lr=False, config='torchdistill/configs/sample/glue/cola/ce/bert_base_uncased.yaml', log='log/glue/cola/ce/bert_base_uncased.txt', private_output='leaderboard/glue/standard/bert_base_uncased/', seed=None, student_only=False, task_name='cola', test_only=False, world_size=1) 2021-05-27 20:30:20,901 INFO __main__ Distributed environment: NO Num processes: 1 Process index: 0 Local process index: 0 Device: cuda Use FP16 precision: True 2021-05-27 20:30:25,701 WARNING datasets.builder Reusing dataset glue (/root/.cache/huggingface/datasets/glue/cola/1.0.0/dacbe3125aa31d7f70367a07a8a9e72a5a0bfeb5fc42e75c9db75b96da6053ad) 2021-05-27 20:30:26,821 INFO __main__ Start training 2021-05-27 20:30:26,821 INFO torchdistill.models.util [student model] 2021-05-27 20:30:26,822 INFO torchdistill.models.util Using the original student model 2021-05-27 20:30:26,822 INFO torchdistill.core.training Loss = 1.0 * OrgLoss 2021-05-27 20:30:29,248 INFO torchdistill.misc.log Epoch: [0] [ 0/535] eta: 0:00:41 lr: 4.9968847352024925e-05 sample/s: 54.10643773502151 loss: 0.6693 (0.6693) time: 0.0775 data: 0.0035 max mem: 842 2021-05-27 20:30:33,369 INFO torchdistill.misc.log Epoch: [0] [ 50/535] eta: 0:00:39 lr: 4.841121495327103e-05 sample/s: 48.982427046990644 loss: 0.5547 (0.5976) time: 0.0834 data: 0.0016 max mem: 2142 2021-05-27 20:30:37,481 INFO torchdistill.misc.log Epoch: [0] [100/535] eta: 0:00:35 lr: 4.685358255451713e-05 sample/s: 49.044858059103305 loss: 0.5561 (0.5635) time: 0.0824 data: 0.0017 max mem: 2142 2021-05-27 20:30:41,654 INFO torchdistill.misc.log Epoch: [0] [150/535] eta: 0:00:31 lr: 4.529595015576324e-05 sample/s: 49.125278535015624 loss: 0.5053 (0.5476) time: 0.0832 data: 0.0018 max mem: 2142 2021-05-27 20:30:45,773 INFO torchdistill.misc.log Epoch: [0] [200/535] eta: 0:00:27 lr: 4.373831775700935e-05 sample/s: 44.87794543669635 loss: 0.3340 (0.5200) time: 0.0836 data: 0.0017 max mem: 2142 2021-05-27 20:30:49,874 INFO torchdistill.misc.log Epoch: [0] [250/535] eta: 0:00:23 lr: 4.218068535825546e-05 sample/s: 44.73637953837621 loss: 0.4407 (0.5087) time: 0.0827 data: 0.0017 max mem: 2142 2021-05-27 20:30:53,937 INFO torchdistill.misc.log Epoch: [0] [300/535] eta: 0:00:19 lr: 4.0623052959501565e-05 sample/s: 48.89948002891319 loss: 0.4557 (0.4978) time: 0.0814 data: 0.0016 max mem: 2142 2021-05-27 20:30:58,107 INFO torchdistill.misc.log Epoch: [0] [350/535] eta: 0:00:15 lr: 3.9065420560747665e-05 sample/s: 58.53960278580301 loss: 0.4266 (0.4919) time: 0.0810 data: 0.0017 max mem: 2298 2021-05-27 20:31:02,230 INFO torchdistill.misc.log Epoch: [0] [400/535] eta: 0:00:11 lr: 3.750778816199377e-05 sample/s: 49.29256081795746 loss: 0.4043 (0.4870) time: 0.0819 data: 0.0017 max mem: 2298 2021-05-27 20:31:06,351 INFO torchdistill.misc.log Epoch: [0] [450/535] eta: 0:00:07 lr: 3.595015576323988e-05 sample/s: 49.25204321277595 loss: 0.3922 (0.4763) time: 0.0816 data: 0.0016 max mem: 2298 2021-05-27 20:31:10,480 INFO torchdistill.misc.log Epoch: [0] [500/535] eta: 0:00:02 lr: 3.4392523364485985e-05 sample/s: 49.238744819975814 loss: 0.4247 (0.4715) time: 0.0816 data: 0.0017 max mem: 2298 2021-05-27 20:31:13,221 INFO torchdistill.misc.log Epoch: [0] Total time: 0:00:44 2021-05-27 20:31:14,429 INFO /usr/local/lib/python3.7/dist-packages/datasets/metric.py Removing /root/.cache/huggingface/metrics/glue/cola/default_experiment-1-0.arrow 2021-05-27 20:31:14,429 INFO __main__ Validation: matthews_correlation = 0.5943825670451766 2021-05-27 20:31:14,429 INFO __main__ Updating ckpt at ./resource/ckpt/glue/cola/ce/cola-bert-base-uncased 2021-05-27 20:31:15,521 INFO torchdistill.misc.log Epoch: [1] [ 0/535] eta: 0:00:39 lr: 3.330218068535826e-05 sample/s: 56.5559720611634 loss: 0.3736 (0.3736) time: 0.0732 data: 0.0025 max mem: 2298 2021-05-27 20:31:19,586 INFO torchdistill.misc.log Epoch: [1] [ 50/535] eta: 0:00:39 lr: 3.1744548286604364e-05 sample/s: 56.025459414874256 loss: 0.2157 (0.2235) time: 0.0829 data: 0.0016 max mem: 2298 2021-05-27 20:31:23,717 INFO torchdistill.misc.log Epoch: [1] [100/535] eta: 0:00:35 lr: 3.018691588785047e-05 sample/s: 49.288650859169124 loss: 0.1558 (0.2244) time: 0.0834 data: 0.0016 max mem: 2298 2021-05-27 20:31:27,825 INFO torchdistill.misc.log Epoch: [1] [150/535] eta: 0:00:31 lr: 2.8629283489096577e-05 sample/s: 49.013336293708136 loss: 0.1865 (0.2305) time: 0.0832 data: 0.0016 max mem: 2298 2021-05-27 20:31:31,934 INFO torchdistill.misc.log Epoch: [1] [200/535] eta: 0:00:27 lr: 2.707165109034268e-05 sample/s: 58.04642408599769 loss: 0.2078 (0.2373) time: 0.0803 data: 0.0016 max mem: 2298 2021-05-27 20:31:36,015 INFO torchdistill.misc.log Epoch: [1] [250/535] eta: 0:00:23 lr: 2.5514018691588787e-05 sample/s: 49.34460780997703 loss: 0.1068 (0.2396) time: 0.0819 data: 0.0016 max mem: 2298 2021-05-27 20:31:40,185 INFO torchdistill.misc.log Epoch: [1] [300/535] eta: 0:00:19 lr: 2.3956386292834894e-05 sample/s: 49.097532424966055 loss: 0.1558 (0.2320) time: 0.0838 data: 0.0016 max mem: 2302 2021-05-27 20:31:44,209 INFO torchdistill.misc.log Epoch: [1] [350/535] eta: 0:00:15 lr: 2.2398753894080997e-05 sample/s: 49.345623746301406 loss: 0.2053 (0.2402) time: 0.0821 data: 0.0016 max mem: 2302 2021-05-27 20:31:48,346 INFO torchdistill.misc.log Epoch: [1] [400/535] eta: 0:00:11 lr: 2.0841121495327104e-05 sample/s: 48.23575615912966 loss: 0.2332 (0.2399) time: 0.0819 data: 0.0017 max mem: 2302 2021-05-27 20:31:52,463 INFO torchdistill.misc.log Epoch: [1] [450/535] eta: 0:00:06 lr: 1.928348909657321e-05 sample/s: 44.758697780635316 loss: 0.1631 (0.2416) time: 0.0818 data: 0.0017 max mem: 2302 2021-05-27 20:31:56,594 INFO torchdistill.misc.log Epoch: [1] [500/535] eta: 0:00:02 lr: 1.7725856697819314e-05 sample/s: 44.85514837005489 loss: 0.2411 (0.2436) time: 0.0816 data: 0.0017 max mem: 2302 2021-05-27 20:31:59,372 INFO torchdistill.misc.log Epoch: [1] Total time: 0:00:43 2021-05-27 20:32:00,587 INFO /usr/local/lib/python3.7/dist-packages/datasets/metric.py Removing /root/.cache/huggingface/metrics/glue/cola/default_experiment-1-0.arrow 2021-05-27 20:32:00,587 INFO __main__ Validation: matthews_correlation = 0.5998094084751417 2021-05-27 20:32:00,588 INFO __main__ Updating ckpt at ./resource/ckpt/glue/cola/ce/cola-bert-base-uncased 2021-05-27 20:32:01,635 INFO torchdistill.misc.log Epoch: [2] [ 0/535] eta: 0:00:45 lr: 1.663551401869159e-05 sample/s: 48.80772442042119 loss: 0.0228 (0.0228) time: 0.0846 data: 0.0026 max mem: 2302 2021-05-27 20:32:05,729 INFO torchdistill.misc.log Epoch: [2] [ 50/535] eta: 0:00:39 lr: 1.5077881619937695e-05 sample/s: 49.13031357252463 loss: 0.0390 (0.1286) time: 0.0790 data: 0.0016 max mem: 2302 2021-05-27 20:32:09,836 INFO torchdistill.misc.log Epoch: [2] [100/535] eta: 0:00:35 lr: 1.3520249221183801e-05 sample/s: 49.299947988445794 loss: 0.0097 (0.1387) time: 0.0811 data: 0.0016 max mem: 2302 2021-05-27 20:32:14,007 INFO torchdistill.misc.log Epoch: [2] [150/535] eta: 0:00:31 lr: 1.1962616822429908e-05 sample/s: 49.242935929533935 loss: 0.0422 (0.1355) time: 0.0848 data: 0.0017 max mem: 2302 2021-05-27 20:32:18,024 INFO torchdistill.misc.log Epoch: [2] [200/535] eta: 0:00:27 lr: 1.0404984423676013e-05 sample/s: 49.29676109434639 loss: 0.0012 (0.1284) time: 0.0793 data: 0.0016 max mem: 2302 2021-05-27 20:32:22,137 INFO torchdistill.misc.log Epoch: [2] [250/535] eta: 0:00:23 lr: 8.84735202492212e-06 sample/s: 49.21852190851698 loss: 0.0336 (0.1363) time: 0.0818 data: 0.0018 max mem: 2302 2021-05-27 20:32:26,212 INFO torchdistill.misc.log Epoch: [2] [300/535] eta: 0:00:19 lr: 7.289719626168225e-06 sample/s: 58.12827762071068 loss: 0.0005 (0.1429) time: 0.0802 data: 0.0016 max mem: 2302 2021-05-27 20:32:30,347 INFO torchdistill.misc.log Epoch: [2] [350/535] eta: 0:00:15 lr: 5.7320872274143305e-06 sample/s: 49.27055727845127 loss: 0.0002 (0.1483) time: 0.0836 data: 0.0017 max mem: 2302 2021-05-27 20:32:34,500 INFO torchdistill.misc.log Epoch: [2] [400/535] eta: 0:00:11 lr: 4.174454828660436e-06 sample/s: 49.26288960991294 loss: 0.0004 (0.1554) time: 0.0837 data: 0.0017 max mem: 2302 2021-05-27 20:32:38,676 INFO torchdistill.misc.log Epoch: [2] [450/535] eta: 0:00:06 lr: 2.6168224299065425e-06 sample/s: 58.12908322361583 loss: 0.0001 (0.1709) time: 0.0835 data: 0.0016 max mem: 2302 2021-05-27 20:32:42,778 INFO torchdistill.misc.log Epoch: [2] [500/535] eta: 0:00:02 lr: 1.059190031152648e-06 sample/s: 48.95069980772429 loss: 0.0002 (0.1783) time: 0.0830 data: 0.0016 max mem: 2302 2021-05-27 20:32:45,536 INFO torchdistill.misc.log Epoch: [2] Total time: 0:00:43 2021-05-27 20:32:46,745 INFO /usr/local/lib/python3.7/dist-packages/datasets/metric.py Removing /root/.cache/huggingface/metrics/glue/cola/default_experiment-1-0.arrow 2021-05-27 20:32:46,745 INFO __main__ Validation: matthews_correlation = 0.6104966084654571 2021-05-27 20:32:46,746 INFO __main__ Updating ckpt at ./resource/ckpt/glue/cola/ce/cola-bert-base-uncased 2021-05-27 20:32:51,744 INFO __main__ [Student: bert-base-uncased] 2021-05-27 20:32:52,965 INFO /usr/local/lib/python3.7/dist-packages/datasets/metric.py Removing /root/.cache/huggingface/metrics/glue/cola/default_experiment-1-0.arrow 2021-05-27 20:32:52,965 INFO __main__ Test: matthews_correlation = 0.6104966084654571 2021-05-27 20:32:52,966 INFO __main__ Start prediction for private dataset(s) 2021-05-27 20:32:52,966 INFO __main__ cola/test: 1063 samples