File size: 6,396 Bytes
18e5eba
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
2021-05-29 21:08:05,621	INFO	__main__	Namespace(adjust_lr=False, config='torchdistill/configs/sample/glue/rte/ce/bert_base_uncased.yaml', log='log/glue/rte/ce/bert_base_uncased.txt', private_output='leaderboard/glue/standard/bert_base_uncased/', seed=None, student_only=False, task_name='rte', test_only=False, world_size=1)
2021-05-29 21:08:05,650	INFO	__main__	Distributed environment: NO
Num processes: 1
Process index: 0
Local process index: 0
Device: cuda
Use FP16 precision: True

2021-05-29 21:08:10,724	WARNING	datasets.builder	Reusing dataset glue (/root/.cache/huggingface/datasets/glue/rte/1.0.0/dacbe3125aa31d7f70367a07a8a9e72a5a0bfeb5fc42e75c9db75b96da6053ad)
2021-05-29 21:08:12,480	INFO	__main__	Start training
2021-05-29 21:08:12,480	INFO	torchdistill.models.util	[student model]
2021-05-29 21:08:12,481	INFO	torchdistill.models.util	Using the original student model
2021-05-29 21:08:12,481	INFO	torchdistill.core.training	Loss = 1.0 * OrgLoss
2021-05-29 21:08:15,195	INFO	torchdistill.misc.log	Epoch: [0]  [  0/156]  eta: 0:00:38  lr: 4.9919871794871795e-05  sample/s: 16.491046863772947  loss: 0.7014 (0.7014)  time: 0.2471  data: 0.0046  max mem: 1900
2021-05-29 21:08:26,519	INFO	torchdistill.misc.log	Epoch: [0]  [ 50/156]  eta: 0:00:24  lr: 4.591346153846154e-05  sample/s: 17.636260618464803  loss: 0.6778 (0.6873)  time: 0.2286  data: 0.0028  max mem: 3189
2021-05-29 21:08:37,850	INFO	torchdistill.misc.log	Epoch: [0]  [100/156]  eta: 0:00:12  lr: 4.190705128205128e-05  sample/s: 17.63246088265216  loss: 0.6690 (0.6821)  time: 0.2286  data: 0.0028  max mem: 3189
2021-05-29 21:08:49,200	INFO	torchdistill.misc.log	Epoch: [0]  [150/156]  eta: 0:00:01  lr: 3.790064102564103e-05  sample/s: 17.644718589450324  loss: 0.6526 (0.6783)  time: 0.2262  data: 0.0027  max mem: 3189
2021-05-29 21:08:50,282	INFO	torchdistill.misc.log	Epoch: [0] Total time: 0:00:35
2021-05-29 21:08:51,602	INFO	/usr/local/lib/python3.7/dist-packages/datasets/metric.py	Removing /root/.cache/huggingface/metrics/glue/rte/default_experiment-1-0.arrow
2021-05-29 21:08:51,603	INFO	__main__	Validation: accuracy = 0.6101083032490975
2021-05-29 21:08:51,603	INFO	__main__	Updating ckpt at ./resource/ckpt/glue/rte/ce/rte-bert-base-uncased
2021-05-29 21:08:52,924	INFO	torchdistill.misc.log	Epoch: [1]  [  0/156]  eta: 0:00:35  lr: 3.7419871794871796e-05  sample/s: 17.58968347993835  loss: 0.6698 (0.6698)  time: 0.2306  data: 0.0032  max mem: 3189
2021-05-29 21:09:04,238	INFO	torchdistill.misc.log	Epoch: [1]  [ 50/156]  eta: 0:00:23  lr: 3.3413461538461536e-05  sample/s: 17.62684768527809  loss: 0.5938 (0.6194)  time: 0.2288  data: 0.0027  max mem: 3189
2021-05-29 21:09:15,629	INFO	torchdistill.misc.log	Epoch: [1]  [100/156]  eta: 0:00:12  lr: 2.9407051282051283e-05  sample/s: 17.62131209182248  loss: 0.6079 (0.6134)  time: 0.2296  data: 0.0027  max mem: 3189
2021-05-29 21:09:27,017	INFO	torchdistill.misc.log	Epoch: [1]  [150/156]  eta: 0:00:01  lr: 2.5400641025641026e-05  sample/s: 17.635037110148975  loss: 0.5751 (0.6014)  time: 0.2262  data: 0.0026  max mem: 3189
2021-05-29 21:09:28,078	INFO	torchdistill.misc.log	Epoch: [1] Total time: 0:00:35
2021-05-29 21:09:29,398	INFO	/usr/local/lib/python3.7/dist-packages/datasets/metric.py	Removing /root/.cache/huggingface/metrics/glue/rte/default_experiment-1-0.arrow
2021-05-29 21:09:29,398	INFO	__main__	Validation: accuracy = 0.6678700361010831
2021-05-29 21:09:29,399	INFO	__main__	Updating ckpt at ./resource/ckpt/glue/rte/ce/rte-bert-base-uncased
2021-05-29 21:09:30,853	INFO	torchdistill.misc.log	Epoch: [2]  [  0/156]  eta: 0:00:36  lr: 2.4919871794871797e-05  sample/s: 17.44987373264604  loss: 0.5368 (0.5368)  time: 0.2326  data: 0.0033  max mem: 3189
2021-05-29 21:09:42,217	INFO	torchdistill.misc.log	Epoch: [2]  [ 50/156]  eta: 0:00:24  lr: 2.091346153846154e-05  sample/s: 17.61959102743042  loss: 0.4061 (0.4451)  time: 0.2270  data: 0.0026  max mem: 3189
2021-05-29 21:09:53,636	INFO	torchdistill.misc.log	Epoch: [2]  [100/156]  eta: 0:00:12  lr: 1.6907051282051284e-05  sample/s: 17.60167862688321  loss: 0.3714 (0.4380)  time: 0.2293  data: 0.0028  max mem: 3189
2021-05-29 21:10:05,060	INFO	torchdistill.misc.log	Epoch: [2]  [150/156]  eta: 0:00:01  lr: 1.2900641025641028e-05  sample/s: 17.62479225977246  loss: 0.4438 (0.4566)  time: 0.2286  data: 0.0028  max mem: 3189
2021-05-29 21:10:06,144	INFO	torchdistill.misc.log	Epoch: [2] Total time: 0:00:35
2021-05-29 21:10:07,463	INFO	/usr/local/lib/python3.7/dist-packages/datasets/metric.py	Removing /root/.cache/huggingface/metrics/glue/rte/default_experiment-1-0.arrow
2021-05-29 21:10:07,463	INFO	__main__	Validation: accuracy = 0.6642599277978339
2021-05-29 21:10:07,694	INFO	torchdistill.misc.log	Epoch: [3]  [  0/156]  eta: 0:00:35  lr: 1.2419871794871796e-05  sample/s: 17.64061842835753  loss: 0.6059 (0.6059)  time: 0.2297  data: 0.0029  max mem: 3189
2021-05-29 21:10:19,117	INFO	torchdistill.misc.log	Epoch: [3]  [ 50/156]  eta: 0:00:24  lr: 8.41346153846154e-06  sample/s: 17.56041029935106  loss: 0.2700 (0.2802)  time: 0.2287  data: 0.0027  max mem: 3189
2021-05-29 21:10:30,490	INFO	torchdistill.misc.log	Epoch: [3]  [100/156]  eta: 0:00:12  lr: 4.4070512820512826e-06  sample/s: 18.516447386573457  loss: 0.2417 (0.2638)  time: 0.2275  data: 0.0028  max mem: 3189
2021-05-29 21:10:41,907	INFO	torchdistill.misc.log	Epoch: [3]  [150/156]  eta: 0:00:01  lr: 4.006410256410256e-07  sample/s: 18.52875802484679  loss: 0.1854 (0.2611)  time: 0.2272  data: 0.0026  max mem: 3189
2021-05-29 21:10:42,988	INFO	torchdistill.misc.log	Epoch: [3] Total time: 0:00:35
2021-05-29 21:10:44,306	INFO	/usr/local/lib/python3.7/dist-packages/datasets/metric.py	Removing /root/.cache/huggingface/metrics/glue/rte/default_experiment-1-0.arrow
2021-05-29 21:10:44,307	INFO	__main__	Validation: accuracy = 0.6750902527075813
2021-05-29 21:10:44,307	INFO	__main__	Updating ckpt at ./resource/ckpt/glue/rte/ce/rte-bert-base-uncased
2021-05-29 21:10:49,239	INFO	__main__	[Student: bert-base-uncased]
2021-05-29 21:10:50,573	INFO	/usr/local/lib/python3.7/dist-packages/datasets/metric.py	Removing /root/.cache/huggingface/metrics/glue/rte/default_experiment-1-0.arrow
2021-05-29 21:10:50,574	INFO	__main__	Test: accuracy = 0.6750902527075813
2021-05-29 21:10:50,574	INFO	__main__	Start prediction for private dataset(s)
2021-05-29 21:10:50,575	INFO	__main__	rte/test: 3000 samples