| [2024-01-18 06:44:35,893][__main__][INFO] - Using GPU for training. |
| [2024-01-18 06:44:37,291][__main__][INFO] - Reconstruction a few images with ADMM... |
| [2024-01-18 06:44:38,300][__main__][INFO] - Train test size : 24000 |
| [2024-01-18 06:44:38,300][__main__][INFO] - Test test size : 999 |
| [2024-01-18 06:44:38,579][__main__][INFO] - Training model with 32641041 parameters |
| [2024-01-18 06:44:38,579][__main__][INFO] - Setup time : 0.2792165279388428 s |
| [2024-01-18 06:44:38,579][__main__][INFO] - PSF shape : torch.Size([1, 135, 240, 3]) |
| [2024-01-18 06:44:38,579][__main__][INFO] - Results saved in /home/bezzam/LenslessPiCam/outputs/2024-01-18/06-44-35 |
| [2024-01-18 06:45:07,928][__main__][INFO] - Epoch 0 with learning rate [0.0001] |
| [2024-01-18 07:11:08,698][__main__][INFO] - loss : 0.28403858492771683 |
| [2024-01-18 07:11:35,762][__main__][INFO] - Epoch 1 with learning rate [0.0001] |
| [2024-01-18 07:37:37,643][__main__][INFO] - loss : 0.24226459578673085 |
| [2024-01-18 07:38:05,874][__main__][INFO] - Epoch 2 with learning rate [0.0001] |
| [2024-01-18 08:04:03,905][__main__][INFO] - loss : 0.22654045476019352 |
| [2024-01-18 08:04:31,739][__main__][INFO] - Epoch 3 with learning rate [0.0001] |
| [2024-01-18 08:30:31,006][__main__][INFO] - loss : 0.21702255016565325 |
| [2024-01-18 08:30:59,603][__main__][INFO] - Epoch 4 with learning rate [0.0001] |
| [2024-01-18 08:56:57,578][__main__][INFO] - loss : 0.21046756836771943 |
| [2024-01-18 08:57:25,391][__main__][INFO] - Epoch 5 with learning rate [0.0001] |
| [2024-01-18 09:23:27,004][__main__][INFO] - loss : 0.20381094432373875 |
| [2024-01-18 09:24:04,044][__main__][INFO] - Epoch 6 with learning rate [0.0001] |
| [2024-01-18 09:51:01,515][__main__][INFO] - loss : 0.1981496449410916 |
| [2024-01-18 09:51:34,179][__main__][INFO] - Epoch 7 with learning rate [0.0001] |
| [2024-01-18 10:18:45,966][__main__][INFO] - loss : 0.19299530608455365 |
| [2024-01-18 10:19:16,004][__main__][INFO] - Epoch 8 with learning rate [0.0001] |
| [2024-01-18 10:46:19,353][__main__][INFO] - loss : 0.18855441852410645 |
| [2024-01-18 10:46:52,732][__main__][INFO] - Epoch 9 with learning rate [0.0001] |
| [2024-01-18 11:14:00,377][__main__][INFO] - loss : 0.18454969284435127 |
| [2024-01-18 11:14:31,421][__main__][INFO] - Epoch 10 with learning rate [0.0001] |
| [2024-01-18 11:41:44,850][__main__][INFO] - loss : 0.18096771164735112 |
| [2024-01-18 11:42:17,378][__main__][INFO] - Epoch 11 with learning rate [0.0001] |
| [2024-01-18 12:09:23,027][__main__][INFO] - loss : 0.17694293196499367 |
| [2024-01-18 12:09:53,444][__main__][INFO] - Epoch 12 with learning rate [0.0001] |
| [2024-01-18 12:37:31,037][__main__][INFO] - loss : 0.17335985261201864 |
| [2024-01-18 12:38:03,582][__main__][INFO] - Epoch 13 with learning rate [0.0001] |
| [2024-01-18 13:05:07,056][__main__][INFO] - loss : 0.17004039297997942 |
| [2024-01-18 13:05:38,116][__main__][INFO] - Epoch 14 with learning rate [0.0001] |
| [2024-01-18 13:32:42,724][__main__][INFO] - loss : 0.1675690046499175 |
| [2024-01-18 13:33:14,867][__main__][INFO] - Epoch 15 with learning rate [0.0001] |
| [2024-01-18 14:00:30,935][__main__][INFO] - loss : 0.16459622901678067 |
| [2024-01-18 14:01:05,146][__main__][INFO] - Epoch 16 with learning rate [0.0001] |
| [2024-01-18 14:28:52,898][__main__][INFO] - loss : 0.16196364759157125 |
| [2024-01-18 14:29:26,848][__main__][INFO] - Epoch 17 with learning rate [0.0001] |
| [2024-01-18 14:56:50,753][__main__][INFO] - loss : 0.1594320773904519 |
| [2024-01-18 14:57:24,324][__main__][INFO] - Epoch 18 with learning rate [0.0001] |
| [2024-01-18 15:24:48,336][__main__][INFO] - loss : 0.15721215479572612 |
| [2024-01-18 15:25:21,655][__main__][INFO] - Epoch 19 with learning rate [0.0001] |
| [2024-01-18 15:52:50,980][__main__][INFO] - loss : 0.15493175284812907 |
| [2024-01-18 15:53:25,285][__main__][INFO] - Epoch 20 with learning rate [0.0001] |
| [2024-01-18 16:21:04,484][__main__][INFO] - loss : 0.15293625986576093 |
| [2024-01-18 16:21:39,666][__main__][INFO] - Epoch 21 with learning rate [0.0001] |
| [2024-01-18 16:49:28,095][__main__][INFO] - loss : 0.15134255375713118 |
| [2024-01-18 16:50:02,184][__main__][INFO] - Epoch 22 with learning rate [0.0001] |
| [2024-01-18 17:17:47,362][__main__][INFO] - loss : 0.1492300958236059 |
| [2024-01-18 17:18:22,799][__main__][INFO] - Epoch 23 with learning rate [0.0001] |
| [2024-01-18 17:45:58,159][__main__][INFO] - loss : 0.14769378089159732 |
| [2024-01-18 17:46:30,903][__main__][INFO] - Epoch 24 with learning rate [0.0001] |
| [2024-01-18 18:14:24,192][__main__][INFO] - loss : 0.14613072758913034 |
| [2024-01-18 18:14:58,335][__main__][INFO] - Train time [hour] : 11.505248521169026 h |
| [2024-01-18 18:14:58,335][__main__][INFO] - Results saved in /home/bezzam/LenslessPiCam/outputs/2024-01-18/06-44-35 |
|
|
