File size: 8,686 Bytes
832b0e3
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
2021-05-27 22:43:02,432	INFO	__main__	Namespace(adjust_lr=False, config='torchdistill/configs/sample/glue/mrpc/ce/bert_base_uncased.yaml', log='log/glue/mrpc/ce/bert_base_uncased.txt', private_output='leaderboard/glue/standard/bert_base_uncased/', seed=None, student_only=False, task_name='mrpc', test_only=False, world_size=1)
2021-05-27 22:43:02,460	INFO	__main__	Distributed environment: NO
Num processes: 1
Process index: 0
Local process index: 0
Device: cuda
Use FP16 precision: True

2021-05-27 22:43:07,596	WARNING	datasets.builder	Reusing dataset glue (/root/.cache/huggingface/datasets/glue/mrpc/1.0.0/dacbe3125aa31d7f70367a07a8a9e72a5a0bfeb5fc42e75c9db75b96da6053ad)
2021-05-27 22:43:09,157	INFO	__main__	Start training
2021-05-27 22:43:09,157	INFO	torchdistill.models.util	[student model]
2021-05-27 22:43:09,158	INFO	torchdistill.models.util	Using the original student model
2021-05-27 22:43:09,158	INFO	torchdistill.core.training	Loss = 1.0 * OrgLoss
2021-05-27 22:43:11,724	INFO	torchdistill.misc.log	Epoch: [0]  [  0/230]  eta: 0:00:38  lr: 4.9956521739130436e-05  sample/s: 24.35787334127007  loss: 0.6288 (0.6288)  time: 0.1682  data: 0.0040  max mem: 1883
2021-05-27 22:43:19,858	INFO	torchdistill.misc.log	Epoch: [0]  [ 50/230]  eta: 0:00:29  lr: 4.7782608695652175e-05  sample/s: 25.151060327587707  loss: 0.5948 (0.6191)  time: 0.1625  data: 0.0024  max mem: 2791
2021-05-27 22:43:27,829	INFO	torchdistill.misc.log	Epoch: [0]  [100/230]  eta: 0:00:20  lr: 4.5608695652173914e-05  sample/s: 26.44901099282227  loss: 0.5307 (0.5978)  time: 0.1588  data: 0.0023  max mem: 2880
2021-05-27 22:43:35,885	INFO	torchdistill.misc.log	Epoch: [0]  [150/230]  eta: 0:00:12  lr: 4.3434782608695654e-05  sample/s: 27.124775269999354  loss: 0.5620 (0.5827)  time: 0.1594  data: 0.0023  max mem: 2880
2021-05-27 22:43:43,849	INFO	torchdistill.misc.log	Epoch: [0]  [200/230]  eta: 0:00:04  lr: 4.126086956521739e-05  sample/s: 25.07550204164599  loss: 0.5060 (0.5631)  time: 0.1580  data: 0.0023  max mem: 2880
2021-05-27 22:43:48,455	INFO	torchdistill.misc.log	Epoch: [0] Total time: 0:00:36
2021-05-27 22:43:49,705	INFO	/usr/local/lib/python3.7/dist-packages/datasets/metric.py	Removing /root/.cache/huggingface/metrics/glue/mrpc/default_experiment-1-0.arrow
2021-05-27 22:43:49,706	INFO	__main__	Validation: accuracy = 0.8112745098039216, f1 = 0.8701517706576728
2021-05-27 22:43:49,706	INFO	__main__	Updating ckpt at ./resource/ckpt/glue/mrpc/ce/mrpc-bert-base-uncased
2021-05-27 22:43:50,994	INFO	torchdistill.misc.log	Epoch: [1]  [  0/230]  eta: 0:00:35  lr: 3.995652173913044e-05  sample/s: 27.032513683654216  loss: 0.5486 (0.5486)  time: 0.1525  data: 0.0045  max mem: 2880
2021-05-27 22:43:58,967	INFO	torchdistill.misc.log	Epoch: [1]  [ 50/230]  eta: 0:00:28  lr: 3.7782608695652176e-05  sample/s: 29.77206923598369  loss: 0.3705 (0.3844)  time: 0.1589  data: 0.0023  max mem: 2880
2021-05-27 22:44:06,974	INFO	torchdistill.misc.log	Epoch: [1]  [100/230]  eta: 0:00:20  lr: 3.5608695652173915e-05  sample/s: 25.122813742872204  loss: 0.3969 (0.3978)  time: 0.1616  data: 0.0023  max mem: 2880
2021-05-27 22:44:15,046	INFO	torchdistill.misc.log	Epoch: [1]  [150/230]  eta: 0:00:12  lr: 3.3434782608695655e-05  sample/s: 23.426238389660583  loss: 0.3003 (0.3903)  time: 0.1623  data: 0.0024  max mem: 2880
2021-05-27 22:44:23,128	INFO	torchdistill.misc.log	Epoch: [1]  [200/230]  eta: 0:00:04  lr: 3.1260869565217394e-05  sample/s: 25.108113002264297  loss: 0.2929 (0.3789)  time: 0.1637  data: 0.0024  max mem: 2880
2021-05-27 22:44:27,685	INFO	torchdistill.misc.log	Epoch: [1] Total time: 0:00:36
2021-05-27 22:44:28,945	INFO	/usr/local/lib/python3.7/dist-packages/datasets/metric.py	Removing /root/.cache/huggingface/metrics/glue/mrpc/default_experiment-1-0.arrow
2021-05-27 22:44:28,946	INFO	__main__	Validation: accuracy = 0.8725490196078431, f1 = 0.9068100358422939
2021-05-27 22:44:28,946	INFO	__main__	Updating ckpt at ./resource/ckpt/glue/mrpc/ce/mrpc-bert-base-uncased
2021-05-27 22:44:30,399	INFO	torchdistill.misc.log	Epoch: [2]  [  0/230]  eta: 0:00:38  lr: 2.9956521739130438e-05  sample/s: 24.807797861266206  loss: 0.0967 (0.0967)  time: 0.1670  data: 0.0058  max mem: 2880
2021-05-27 22:44:38,497	INFO	torchdistill.misc.log	Epoch: [2]  [ 50/230]  eta: 0:00:29  lr: 2.7782608695652174e-05  sample/s: 25.094029980226573  loss: 0.0729 (0.1408)  time: 0.1629  data: 0.0023  max mem: 2880
2021-05-27 22:44:46,653	INFO	torchdistill.misc.log	Epoch: [2]  [100/230]  eta: 0:00:21  lr: 2.5608695652173913e-05  sample/s: 25.213161711112615  loss: 0.0524 (0.1608)  time: 0.1631  data: 0.0024  max mem: 2880
2021-05-27 22:44:54,619	INFO	torchdistill.misc.log	Epoch: [2]  [150/230]  eta: 0:00:12  lr: 2.3434782608695656e-05  sample/s: 25.193095525906795  loss: 0.0695 (0.1629)  time: 0.1553  data: 0.0023  max mem: 2880
2021-05-27 22:45:02,620	INFO	torchdistill.misc.log	Epoch: [2]  [200/230]  eta: 0:00:04  lr: 2.126086956521739e-05  sample/s: 27.233573952721294  loss: 0.1384 (0.1721)  time: 0.1623  data: 0.0023  max mem: 2880
2021-05-27 22:45:07,172	INFO	torchdistill.misc.log	Epoch: [2] Total time: 0:00:36
2021-05-27 22:45:08,428	INFO	/usr/local/lib/python3.7/dist-packages/datasets/metric.py	Removing /root/.cache/huggingface/metrics/glue/mrpc/default_experiment-1-0.arrow
2021-05-27 22:45:08,428	INFO	__main__	Validation: accuracy = 0.8553921568627451, f1 = 0.8984509466437176
2021-05-27 22:45:08,590	INFO	torchdistill.misc.log	Epoch: [3]  [  0/230]  eta: 0:00:37  lr: 1.9956521739130435e-05  sample/s: 25.15935804498244  loss: 0.0007 (0.0007)  time: 0.1614  data: 0.0024  max mem: 2880
2021-05-27 22:45:16,673	INFO	torchdistill.misc.log	Epoch: [3]  [ 50/230]  eta: 0:00:29  lr: 1.7782608695652174e-05  sample/s: 25.001066968676422  loss: 0.0001 (0.0465)  time: 0.1639  data: 0.0023  max mem: 2880
2021-05-27 22:45:24,669	INFO	torchdistill.misc.log	Epoch: [3]  [100/230]  eta: 0:00:20  lr: 1.5608695652173914e-05  sample/s: 27.17050458396156  loss: 0.0000 (0.0732)  time: 0.1560  data: 0.0023  max mem: 2880
2021-05-27 22:45:32,573	INFO	torchdistill.misc.log	Epoch: [3]  [150/230]  eta: 0:00:12  lr: 1.3434782608695653e-05  sample/s: 25.425263387607142  loss: 0.0000 (0.0976)  time: 0.1571  data: 0.0023  max mem: 2880
2021-05-27 22:45:40,580	INFO	torchdistill.misc.log	Epoch: [3]  [200/230]  eta: 0:00:04  lr: 1.126086956521739e-05  sample/s: 22.02089838648496  loss: 0.0000 (0.1179)  time: 0.1596  data: 0.0024  max mem: 2880
2021-05-27 22:45:45,185	INFO	torchdistill.misc.log	Epoch: [3] Total time: 0:00:36
2021-05-27 22:45:46,434	INFO	/usr/local/lib/python3.7/dist-packages/datasets/metric.py	Removing /root/.cache/huggingface/metrics/glue/mrpc/default_experiment-1-0.arrow
2021-05-27 22:45:46,434	INFO	__main__	Validation: accuracy = 0.8112745098039216, f1 = 0.8499025341130605
2021-05-27 22:45:46,608	INFO	torchdistill.misc.log	Epoch: [4]  [  0/230]  eta: 0:00:39  lr: 9.956521739130436e-06  sample/s: 23.514462112027584  loss: 0.3838 (0.3838)  time: 0.1726  data: 0.0025  max mem: 2880
2021-05-27 22:45:54,606	INFO	torchdistill.misc.log	Epoch: [4]  [ 50/230]  eta: 0:00:28  lr: 7.782608695652174e-06  sample/s: 27.12218811178524  loss: 0.0000 (0.0474)  time: 0.1598  data: 0.0023  max mem: 2880
2021-05-27 22:46:02,496	INFO	torchdistill.misc.log	Epoch: [4]  [100/230]  eta: 0:00:20  lr: 5.608695652173914e-06  sample/s: 27.583716001703305  loss: 0.0000 (0.0583)  time: 0.1606  data: 0.0023  max mem: 2880
2021-05-27 22:46:10,378	INFO	torchdistill.misc.log	Epoch: [4]  [150/230]  eta: 0:00:12  lr: 3.4347826086956526e-06  sample/s: 22.05644916380617  loss: 0.0000 (0.0455)  time: 0.1558  data: 0.0024  max mem: 2880
2021-05-27 22:46:18,525	INFO	torchdistill.misc.log	Epoch: [4]  [200/230]  eta: 0:00:04  lr: 1.2608695652173913e-06  sample/s: 23.465589513420824  loss: 0.0000 (0.0659)  time: 0.1637  data: 0.0023  max mem: 2880
2021-05-27 22:46:22,985	INFO	torchdistill.misc.log	Epoch: [4] Total time: 0:00:36
2021-05-27 22:46:24,235	INFO	/usr/local/lib/python3.7/dist-packages/datasets/metric.py	Removing /root/.cache/huggingface/metrics/glue/mrpc/default_experiment-1-0.arrow
2021-05-27 22:46:24,235	INFO	__main__	Validation: accuracy = 0.8480392156862745, f1 = 0.8938356164383561
2021-05-27 22:46:27,993	INFO	__main__	[Student: bert-base-uncased]
2021-05-27 22:46:29,258	INFO	/usr/local/lib/python3.7/dist-packages/datasets/metric.py	Removing /root/.cache/huggingface/metrics/glue/mrpc/default_experiment-1-0.arrow
2021-05-27 22:46:29,259	INFO	__main__	Test: accuracy = 0.8725490196078431, f1 = 0.9068100358422939
2021-05-27 22:46:29,259	INFO	__main__	Start prediction for private dataset(s)
2021-05-27 22:46:29,260	INFO	__main__	mrpc/test: 1725 samples