| [2025-02-28 17:57:10,538][__main__][INFO] - Using GPU for training. Main device : cuda:2 | |
| [2025-02-28 17:57:15,825][__main__][INFO] - PSF shape : torch.Size([1, 135, 240, 3]) | |
| [2025-02-28 17:57:15,836][__main__][INFO] - PSF min : 0.0 | |
| [2025-02-28 17:57:15,842][__main__][INFO] - PSF max : 0.12337537109851837 | |
| [2025-02-28 17:57:15,842][__main__][INFO] - PSF dtype : torch.float32 | |
| [2025-02-28 17:57:15,859][__main__][INFO] - PSF norm : 1.7320507764816284 | |
| [2025-02-28 17:57:15,860][__main__][INFO] - Reconstruction a few images with ADMM... | |
| [2025-02-28 17:57:16,422][__main__][INFO] - Cropped shape : (135, 240, 3) | |
| [2025-02-28 17:57:17,539][__main__][INFO] - Train test size : 24000 | |
| [2025-02-28 17:57:17,540][__main__][INFO] - Test test size : 999 | |
| [2025-02-28 17:57:17,677][__main__][INFO] - Training model with 8089718 parameters | |
| [2025-02-28 17:57:17,678][__main__][INFO] - -- Pre-process model with 4044848 parameters | |
| [2025-02-28 17:57:17,679][__main__][INFO] - -- Post-process model with 4044848 parameters | |
| [2025-02-28 17:57:17,679][__main__][INFO] - Setup time : 0.13936614990234375 s | |
| [2025-02-28 17:57:17,679][__main__][INFO] - PSF shape : torch.Size([1, 135, 240, 3]) | |
| [2025-02-28 17:57:17,679][__main__][INFO] - Results saved in /root/LenslessPiCam/outputs/2025-02-28/17-57-09 | |
| [2025-02-28 17:59:05,332][__main__][INFO] - Epoch 0 with learning rate [0.0, 0.0] | |
| [2025-02-28 20:35:39,196][__main__][INFO] - loss : 0.46096361348529824 | |
| [2025-02-28 20:37:25,865][__main__][INFO] - Epoch 1 with learning rate [8e-05, 8e-05] | |
| [2025-02-28 23:14:34,769][__main__][INFO] - loss : 0.2676947735827411 | |
| [2025-02-28 23:16:21,531][__main__][INFO] - Epoch 2 with learning rate [9.975414512725057e-05, 9.975414512725057e-05] | |
| [2025-03-01 01:53:38,496][__main__][INFO] - loss : 0.2237047298476101 | |
| [2025-03-01 01:55:25,314][__main__][INFO] - Epoch 3 with learning rate [9.86663298624003e-05, 9.86663298624003e-05] | |
| [2025-03-01 04:31:55,108][__main__][INFO] - loss : 0.20493856664809174 | |
| [2025-03-01 04:33:41,683][__main__][INFO] - Epoch 4 with learning rate [9.672822322997305e-05, 9.672822322997305e-05] | |
| [2025-03-01 07:10:27,215][__main__][INFO] - loss : 0.19325304987219377 | |
| [2025-03-01 07:12:13,729][__main__][INFO] - Epoch 5 with learning rate [9.397368756032445e-05, 9.397368756032445e-05] | |
| [2025-03-01 09:49:17,717][__main__][INFO] - loss : 0.1847326516338313 | |
| [2025-03-01 09:51:05,449][__main__][INFO] - Epoch 6 with learning rate [9.045084971874738e-05, 9.045084971874738e-05] | |
| [2025-03-01 12:28:22,027][__main__][INFO] - loss : 0.17733714263451608 | |
| [2025-03-01 12:30:09,367][__main__][INFO] - Epoch 7 with learning rate [8.622126023955446e-05, 8.622126023955446e-05] | |
| [2025-03-01 15:07:20,205][__main__][INFO] - loss : 0.17084609245508905 | |
| [2025-03-01 15:09:07,820][__main__][INFO] - Epoch 8 with learning rate [8.135881792367686e-05, 8.135881792367686e-05] | |
| [2025-03-01 17:46:34,084][__main__][INFO] - loss : 0.16506924781544766 | |
| [2025-03-01 17:48:21,845][__main__][INFO] - Epoch 9 with learning rate [7.594847868906076e-05, 7.594847868906076e-05] | |
| [2025-03-01 20:25:28,977][__main__][INFO] - loss : 0.15971624759988304 | |
| [2025-03-01 20:27:16,370][__main__][INFO] - Epoch 10 with learning rate [7.008477123264848e-05, 7.008477123264848e-05] | |
| [2025-03-01 23:04:22,024][__main__][INFO] - loss : 0.15440590443524246 | |
| [2025-03-01 23:06:09,563][__main__][INFO] - Epoch 11 with learning rate [6.387014543809223e-05, 6.387014543809223e-05] | |
| [2025-03-02 01:43:37,032][__main__][INFO] - loss : 0.14950965759127038 | |
| [2025-03-02 01:45:24,526][__main__][INFO] - Epoch 12 with learning rate [5.74131823855921e-05, 5.74131823855921e-05] | |
| [2025-03-02 04:22:57,344][__main__][INFO] - loss : 0.1446941166844527 | |
| [2025-03-02 04:24:45,038][__main__][INFO] - Epoch 13 with learning rate [5.0826697238317935e-05, 5.0826697238317935e-05] | |
| [2025-03-02 07:02:00,694][__main__][INFO] - loss : 0.14001506880422382 | |
| [2025-03-02 07:03:47,566][__main__][INFO] - Epoch 14 with learning rate [4.4225768151520694e-05, 4.4225768151520694e-05] | |
| [2025-03-02 09:40:30,074][__main__][INFO] - loss : 0.13556548026273949 | |
| [2025-03-02 09:42:17,340][__main__][INFO] - Epoch 15 with learning rate [3.772572564296005e-05, 3.772572564296005e-05] | |
| [2025-03-02 12:19:01,528][__main__][INFO] - loss : 0.13123598493852906 | |
| [2025-03-02 12:20:49,158][__main__][INFO] - Epoch 16 with learning rate [3.144013755408895e-05, 3.144013755408895e-05] | |
| [2025-03-02 14:57:37,849][__main__][INFO] - loss : 0.12723831308260597 | |
| [2025-03-02 14:59:25,363][__main__][INFO] - Epoch 17 with learning rate [2.547882480847461e-05, 2.547882480847461e-05] | |
| [2025-03-02 17:36:06,201][__main__][INFO] - loss : 0.1235241475754106 | |
| [2025-03-02 17:37:52,802][__main__][INFO] - Epoch 18 with learning rate [1.9945942635848748e-05, 1.9945942635848748e-05] | |
| [2025-03-02 20:13:35,065][__main__][INFO] - loss : 0.1201749133223979 | |
| [2025-03-02 20:15:21,064][__main__][INFO] - Epoch 19 with learning rate [1.4938160786375572e-05, 1.4938160786375572e-05] | |
| [2025-03-02 22:50:57,342][__main__][INFO] - loss : 0.11728145546000421 | |
| [2025-03-02 22:52:43,111][__main__][INFO] - Epoch 20 with learning rate [1.0542974530180327e-05, 1.0542974530180327e-05] | |
| [2025-03-03 01:28:24,113][__main__][INFO] - loss : 0.11491350826970342 | |
| [2025-03-03 01:30:10,167][__main__][INFO] - Epoch 21 with learning rate [6.837175952121306e-06, 6.837175952121306e-06] | |
| [2025-03-03 04:05:53,578][__main__][INFO] - loss : 0.11304938979431389 | |
| [2025-03-03 04:07:39,366][__main__][INFO] - Epoch 22 with learning rate [3.885512251130763e-06, 3.885512251130763e-06] | |
| [2025-03-03 06:43:20,001][__main__][INFO] - loss : 0.11173099756116667 | |
| [2025-03-03 06:45:05,426][__main__][INFO] - Epoch 23 with learning rate [1.7395544861325718e-06, 1.7395544861325718e-06] | |
| [2025-03-03 09:20:59,904][__main__][INFO] - loss : 0.11091148214492332 | |
| [2025-03-03 09:22:46,290][__main__][INFO] - Epoch 24 with learning rate [4.367965336512403e-07, 4.367965336512403e-07] | |
| [2025-03-03 11:58:50,940][__main__][INFO] - loss : 0.11052820207566644 | |
| [2025-03-03 12:00:37,496][__main__][INFO] - Train time [hour] : 66.05510537220373 h | |
| [2025-03-03 12:00:37,496][__main__][INFO] - Results saved in /root/LenslessPiCam/outputs/2025-02-28/17-57-09 | |