| [2025-03-06 17:31:50,783][__main__][INFO] - Using GPU for training. Main device : cuda:0 | |
| [2025-03-06 17:31:55,653][__main__][INFO] - PSF shape : torch.Size([1, 135, 240, 3]) | |
| [2025-03-06 17:31:55,666][__main__][INFO] - PSF min : 0.0 | |
| [2025-03-06 17:31:55,671][__main__][INFO] - PSF max : 0.12337537109851837 | |
| [2025-03-06 17:31:55,671][__main__][INFO] - PSF dtype : torch.float32 | |
| [2025-03-06 17:31:55,689][__main__][INFO] - PSF norm : 1.7320507764816284 | |
| [2025-03-06 17:31:55,689][__main__][INFO] - Reconstruction a few images with ADMM... | |
| [2025-03-06 17:31:56,366][__main__][INFO] - Cropped shape : (135, 240, 3) | |
| [2025-03-06 17:31:57,464][__main__][INFO] - Train test size : 24000 | |
| [2025-03-06 17:31:57,465][__main__][INFO] - Test test size : 999 | |
| [2025-03-06 17:31:57,574][__main__][INFO] - Training model with 8092627 parameters | |
| [2025-03-06 17:31:57,574][__main__][INFO] - -- Pre-process model with 3913696 parameters | |
| [2025-03-06 17:31:57,575][__main__][INFO] - -- Post-process model with 4051168 parameters | |
| [2025-03-06 17:31:57,575][__main__][INFO] - -- PSF network model with 127740 parameters | |
| [2025-03-06 17:31:57,576][__main__][INFO] - Setup time : 0.11081218719482422 s | |
| [2025-03-06 17:31:57,576][__main__][INFO] - PSF shape : torch.Size([1, 135, 240, 3]) | |
| [2025-03-06 17:31:57,576][__main__][INFO] - Results saved in /root/LenslessPiCam/outputs/2025-03-06/17-31-49 | |
| [2025-03-06 17:32:46,041][__main__][INFO] - Epoch 0 with learning rate [0.0, 0.0] | |
| [2025-03-06 18:05:43,786][__main__][INFO] - loss : 0.4539545120249171 | |
| [2025-03-06 18:06:31,630][__main__][INFO] - Epoch 1 with learning rate [8e-05, 8e-05] | |
| [2025-03-06 18:39:23,754][__main__][INFO] - loss : 0.2671020127634204 | |
| [2025-03-06 18:40:10,856][__main__][INFO] - Epoch 2 with learning rate [9.975414512725057e-05, 9.975414512725057e-05] | |
| [2025-03-06 19:12:58,415][__main__][INFO] - loss : 0.2237230134829881 | |
| [2025-03-06 19:13:44,064][__main__][INFO] - Epoch 3 with learning rate [9.86663298624003e-05, 9.86663298624003e-05] | |
| [2025-03-06 19:46:35,307][__main__][INFO] - loss : 0.2030943300401169 | |
| [2025-03-06 19:47:20,948][__main__][INFO] - Epoch 4 with learning rate [9.672822322997305e-05, 9.672822322997305e-05] | |
| [2025-03-06 20:20:07,215][__main__][INFO] - loss : 0.19035938401147667 | |
| [2025-03-06 20:20:53,459][__main__][INFO] - Epoch 5 with learning rate [9.397368756032445e-05, 9.397368756032445e-05] | |
| [2025-03-06 20:53:41,084][__main__][INFO] - loss : 0.18135352981711383 | |
| [2025-03-06 20:54:26,984][__main__][INFO] - Epoch 6 with learning rate [9.045084971874738e-05, 9.045084971874738e-05] | |
| [2025-03-06 21:27:12,099][__main__][INFO] - loss : 0.1741235060567658 | |
| [2025-03-06 21:27:58,031][__main__][INFO] - Epoch 7 with learning rate [8.622126023955446e-05, 8.622126023955446e-05] | |
| [2025-03-06 22:00:39,856][__main__][INFO] - loss : 0.1683176435368753 | |
| [2025-03-06 22:01:25,584][__main__][INFO] - Epoch 8 with learning rate [8.135881792367686e-05, 8.135881792367686e-05] | |
| [2025-03-06 22:34:10,832][__main__][INFO] - loss : 0.1631974630393081 | |
| [2025-03-06 22:34:56,459][__main__][INFO] - Epoch 9 with learning rate [7.594847868906076e-05, 7.594847868906076e-05] | |
| [2025-03-06 23:07:44,938][__main__][INFO] - loss : 0.1581276698100069 | |
| [2025-03-06 23:08:31,764][__main__][INFO] - Epoch 10 with learning rate [7.008477123264848e-05, 7.008477123264848e-05] | |
| [2025-03-06 23:41:25,695][__main__][INFO] - loss : 0.15349892479553762 | |
| [2025-03-06 23:42:12,737][__main__][INFO] - Epoch 11 with learning rate [6.387014543809223e-05, 6.387014543809223e-05] | |
| [2025-03-07 00:15:01,457][__main__][INFO] - loss : 0.1488871829062697 | |
| [2025-03-07 00:15:47,085][__main__][INFO] - Epoch 12 with learning rate [5.74131823855921e-05, 5.74131823855921e-05] | |
| [2025-03-07 00:48:38,289][__main__][INFO] - loss : 0.14450448525945409 | |
| [2025-03-07 00:49:23,465][__main__][INFO] - Epoch 13 with learning rate [5.0826697238317935e-05, 5.0826697238317935e-05] | |
| [2025-03-07 01:22:09,668][__main__][INFO] - loss : 0.1402942953482272 | |
| [2025-03-07 01:22:56,038][__main__][INFO] - Epoch 14 with learning rate [4.4225768151520694e-05, 4.4225768151520694e-05] | |
| [2025-03-07 01:55:44,163][__main__][INFO] - loss : 0.13628447369734417 | |
| [2025-03-07 01:56:30,182][__main__][INFO] - Epoch 15 with learning rate [3.772572564296005e-05, 3.772572564296005e-05] | |
| [2025-03-07 02:29:08,231][__main__][INFO] - loss : 0.132504320823898 | |
| [2025-03-07 02:29:53,914][__main__][INFO] - Epoch 16 with learning rate [3.144013755408895e-05, 3.144013755408895e-05] | |
| [2025-03-07 03:02:38,948][__main__][INFO] - loss : 0.1289045843183989 | |
| [2025-03-07 03:03:25,142][__main__][INFO] - Epoch 17 with learning rate [2.547882480847461e-05, 2.547882480847461e-05] | |
| [2025-03-07 03:36:15,688][__main__][INFO] - loss : 0.12564500992186356 | |
| [2025-03-07 03:37:02,115][__main__][INFO] - Epoch 18 with learning rate [1.9945942635848748e-05, 1.9945942635848748e-05] | |
| [2025-03-07 04:09:50,758][__main__][INFO] - loss : 0.12274628370876131 | |
| [2025-03-07 04:10:36,582][__main__][INFO] - Epoch 19 with learning rate [1.4938160786375572e-05, 1.4938160786375572e-05] | |
| [2025-03-07 04:43:22,321][__main__][INFO] - loss : 0.12023946041303382 | |
| [2025-03-07 04:44:08,402][__main__][INFO] - Epoch 20 with learning rate [1.0542974530180327e-05, 1.0542974530180327e-05] | |
| [2025-03-07 05:16:56,167][__main__][INFO] - loss : 0.11816099139054614 | |
| [2025-03-07 05:17:41,368][__main__][INFO] - Epoch 21 with learning rate [6.837175952121306e-06, 6.837175952121306e-06] | |
| [2025-03-07 05:50:36,441][__main__][INFO] - loss : 0.11657386542049565 | |
| [2025-03-07 05:51:22,931][__main__][INFO] - Epoch 22 with learning rate [3.885512251130763e-06, 3.885512251130763e-06] | |
| [2025-03-07 06:24:19,364][__main__][INFO] - loss : 0.11544328222982551 | |
| [2025-03-07 06:25:05,669][__main__][INFO] - Epoch 23 with learning rate [1.7395544861325718e-06, 1.7395544861325718e-06] | |
| [2025-03-07 06:58:07,599][__main__][INFO] - loss : 0.1147212131898851 | |
| [2025-03-07 06:58:53,817][__main__][INFO] - Epoch 24 with learning rate [4.367965336512403e-07, 4.367965336512403e-07] | |
| [2025-03-07 07:31:48,463][__main__][INFO] - loss : 0.11437387950780495 | |
| [2025-03-07 07:32:34,661][__main__][INFO] - Train time [hour] : 14.00991320338514 h | |
| [2025-03-07 07:32:34,661][__main__][INFO] - Results saved in /root/LenslessPiCam/outputs/2025-03-06/17-31-49 | |