File size: 4,634 Bytes
ccbf744 |
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 |
[2024-01-17 20:06:45,733][__main__][INFO] - Using GPU for training.
[2024-01-17 20:06:47,198][__main__][INFO] - Reconstruction a few images with ADMM...
[2024-01-17 20:06:48,262][__main__][INFO] - Train test size : 24000
[2024-01-17 20:06:48,262][__main__][INFO] - Test test size : 999
[2024-01-17 20:06:48,375][__main__][INFO] - Training model with 17010705 parameters
[2024-01-17 20:06:48,376][__main__][INFO] - Setup time : 0.11329770088195801 s
[2024-01-17 20:06:48,376][__main__][INFO] - PSF shape : torch.Size([1, 135, 240, 3])
[2024-01-17 20:06:48,376][__main__][INFO] - Results saved in /home/bezzam/LenslessPiCam/outputs/2024-01-17/20-06-45
[2024-01-17 20:07:23,069][__main__][INFO] - Epoch 0 with learning rate [0.0001]
[2024-01-17 20:33:23,142][__main__][INFO] - loss : 0.36667297692100115
[2024-01-17 20:33:55,814][__main__][INFO] - Epoch 1 with learning rate [0.0001]
[2024-01-17 20:59:54,554][__main__][INFO] - loss : 0.2878954428384699
[2024-01-17 21:00:27,578][__main__][INFO] - Epoch 2 with learning rate [0.0001]
[2024-01-17 21:26:08,478][__main__][INFO] - loss : 0.26462542621791285
[2024-01-17 21:26:40,818][__main__][INFO] - Epoch 3 with learning rate [0.0001]
[2024-01-17 21:52:25,994][__main__][INFO] - loss : 0.2509993498772382
[2024-01-17 21:52:58,753][__main__][INFO] - Epoch 4 with learning rate [0.0001]
[2024-01-17 22:18:53,004][__main__][INFO] - loss : 0.24276707645257278
[2024-01-17 22:19:26,145][__main__][INFO] - Epoch 5 with learning rate [0.0001]
[2024-01-17 22:45:09,006][__main__][INFO] - loss : 0.23589916938543318
[2024-01-17 22:45:41,630][__main__][INFO] - Epoch 6 with learning rate [0.0001]
[2024-01-17 23:11:29,911][__main__][INFO] - loss : 0.23032548744976525
[2024-01-17 23:12:03,028][__main__][INFO] - Epoch 7 with learning rate [0.0001]
[2024-01-17 23:37:46,201][__main__][INFO] - loss : 0.2260298660149175
[2024-01-17 23:38:16,925][__main__][INFO] - Epoch 8 with learning rate [0.0001]
[2024-01-18 00:04:13,568][__main__][INFO] - loss : 0.2212433759719131
[2024-01-18 00:04:47,630][__main__][INFO] - Epoch 9 with learning rate [0.0001]
[2024-01-18 00:30:36,234][__main__][INFO] - loss : 0.21692666750649664
[2024-01-18 00:31:09,535][__main__][INFO] - Epoch 10 with learning rate [0.0001]
[2024-01-18 00:56:54,932][__main__][INFO] - loss : 0.21269827534258345
[2024-01-18 00:57:26,649][__main__][INFO] - Epoch 11 with learning rate [0.0001]
[2024-01-18 01:23:25,489][__main__][INFO] - loss : 0.20919359432657592
[2024-01-18 01:23:59,103][__main__][INFO] - Epoch 12 with learning rate [0.0001]
[2024-01-18 01:49:46,540][__main__][INFO] - loss : 0.20483679519097045
[2024-01-18 01:50:19,357][__main__][INFO] - Epoch 13 with learning rate [0.0001]
[2024-01-18 02:15:44,356][__main__][INFO] - loss : 0.20158936556180315
[2024-01-18 02:16:18,075][__main__][INFO] - Epoch 14 with learning rate [0.0001]
[2024-01-18 02:41:55,475][__main__][INFO] - loss : 0.19783804734051214
[2024-01-18 02:42:27,379][__main__][INFO] - Epoch 15 with learning rate [0.0001]
[2024-01-18 03:08:05,537][__main__][INFO] - loss : 0.19492025763293092
[2024-01-18 03:08:36,388][__main__][INFO] - Epoch 16 with learning rate [0.0001]
[2024-01-18 03:34:16,130][__main__][INFO] - loss : 0.19140285502374169
[2024-01-18 03:34:52,056][__main__][INFO] - Epoch 17 with learning rate [0.0001]
[2024-01-18 04:00:32,084][__main__][INFO] - loss : 0.18805657176673365
[2024-01-18 04:01:02,232][__main__][INFO] - Epoch 18 with learning rate [0.0001]
[2024-01-18 04:26:36,151][__main__][INFO] - loss : 0.185264913658301
[2024-01-18 04:27:09,632][__main__][INFO] - Epoch 19 with learning rate [0.0001]
[2024-01-18 04:52:43,178][__main__][INFO] - loss : 0.18217022760709117
[2024-01-18 04:53:15,053][__main__][INFO] - Epoch 20 with learning rate [0.0001]
[2024-01-18 05:16:37,190][__main__][INFO] - loss : 0.17973323059330423
[2024-01-18 05:17:00,886][__main__][INFO] - Epoch 21 with learning rate [0.0001]
[2024-01-18 05:38:29,839][__main__][INFO] - loss : 0.1770827673673628
[2024-01-18 05:38:53,403][__main__][INFO] - Epoch 22 with learning rate [0.0001]
[2024-01-18 06:00:22,922][__main__][INFO] - loss : 0.17473657063643144
[2024-01-18 06:00:46,463][__main__][INFO] - Epoch 23 with learning rate [0.0001]
[2024-01-18 06:22:15,708][__main__][INFO] - loss : 0.17264075693736475
[2024-01-18 06:22:38,862][__main__][INFO] - Epoch 24 with learning rate [0.0001]
[2024-01-18 06:44:08,292][__main__][INFO] - loss : 0.17019510875393928
[2024-01-18 06:44:31,911][__main__][INFO] - Train time [hour] : 10.628406446443664 h
[2024-01-18 06:44:31,912][__main__][INFO] - Results saved in /home/bezzam/LenslessPiCam/outputs/2024-01-17/20-06-45
|