| [2024-01-27 16:04:47,153][__main__][INFO] - Using GPU for training. | |
| [2024-01-27 16:04:49,168][__main__][INFO] - Reconstruction a few images with ADMM... | |
| [2024-01-27 16:04:50,427][__main__][INFO] - Train test size : 24000 | |
| [2024-01-27 16:04:50,427][__main__][INFO] - Test test size : 999 | |
| [2024-01-27 16:04:50,980][__main__][INFO] - Training model with 34021330 parameters | |
| [2024-01-27 16:04:50,980][__main__][INFO] - Setup time : 0.5530400276184082 s | |
| [2024-01-27 16:04:50,980][__main__][INFO] - PSF shape : torch.Size([1, 135, 240, 3]) | |
| [2024-01-27 16:04:50,980][__main__][INFO] - Results saved in /home/bezzam/LenslessPiCam/outputs/2024-01-27/16-04-47 | |
| [2024-01-27 16:05:21,669][__main__][INFO] - Epoch 0 with learning rate [0.0001] | |
| [2024-01-27 16:31:56,246][__main__][INFO] - loss : 0.2904620627711222 | |
| [2024-01-27 16:32:24,617][__main__][INFO] - Epoch 1 with learning rate [0.0001] | |
| [2024-01-27 16:58:56,452][__main__][INFO] - loss : 0.22702715347210572 | |
| [2024-01-27 16:59:25,164][__main__][INFO] - Epoch 2 with learning rate [0.0001] | |
| [2024-01-27 17:25:54,003][__main__][INFO] - loss : 0.20910989487667875 | |
| [2024-01-27 17:26:21,353][__main__][INFO] - Epoch 3 with learning rate [0.0001] | |
| [2024-01-27 17:52:50,291][__main__][INFO] - loss : 0.19793190091351665 | |
| [2024-01-27 17:53:18,505][__main__][INFO] - Epoch 4 with learning rate [0.0001] | |
| [2024-01-27 18:19:46,649][__main__][INFO] - loss : 0.19041558116177718 | |
| [2024-01-27 18:20:14,007][__main__][INFO] - Epoch 5 with learning rate [0.0001] | |
| [2024-01-27 18:46:44,381][__main__][INFO] - loss : 0.1843203121076023 | |
| [2024-01-27 18:47:11,732][__main__][INFO] - Epoch 6 with learning rate [0.0001] | |
| [2024-01-27 19:13:41,149][__main__][INFO] - loss : 0.17984511188914334 | |
| [2024-01-27 19:14:08,607][__main__][INFO] - Epoch 7 with learning rate [0.0001] | |
| [2024-01-27 19:40:49,725][__main__][INFO] - loss : 0.1747767227689429 | |
| [2024-01-27 19:41:17,730][__main__][INFO] - Epoch 8 with learning rate [0.0001] | |
| [2024-01-27 20:07:45,613][__main__][INFO] - loss : 0.17105758504569524 | |
| [2024-01-27 20:08:13,562][__main__][INFO] - Epoch 9 with learning rate [0.0001] | |
| [2024-01-27 20:34:42,224][__main__][INFO] - loss : 0.16789209530999252 | |
| [2024-01-27 20:35:11,232][__main__][INFO] - Epoch 10 with learning rate [0.0001] | |
| [2024-01-27 21:01:38,963][__main__][INFO] - loss : 0.16390000306318184 | |
| [2024-01-27 21:02:07,254][__main__][INFO] - Epoch 11 with learning rate [0.0001] | |
| [2024-01-27 21:28:35,031][__main__][INFO] - loss : 0.1614447848250468 | |
| [2024-01-27 21:29:03,631][__main__][INFO] - Epoch 12 with learning rate [0.0001] | |
| [2024-01-27 21:55:31,473][__main__][INFO] - loss : 0.15819583714753419 | |
| [2024-01-27 21:55:58,636][__main__][INFO] - Epoch 13 with learning rate [0.0001] | |
| [2024-01-27 22:22:27,409][__main__][INFO] - loss : 0.15576349579791215 | |
| [2024-01-27 22:22:54,695][__main__][INFO] - Epoch 14 with learning rate [0.0001] | |
| [2024-01-27 22:49:22,344][__main__][INFO] - loss : 0.15290123852839085 | |
| [2024-01-27 22:49:50,259][__main__][INFO] - Epoch 15 with learning rate [0.0001] | |
| [2024-01-27 23:16:16,848][__main__][INFO] - loss : 0.15053067220995822 | |
| [2024-01-27 23:16:44,718][__main__][INFO] - Epoch 16 with learning rate [0.0001] | |
| [2024-01-27 23:43:14,301][__main__][INFO] - loss : 0.1479654023870823 | |
| [2024-01-27 23:43:42,591][__main__][INFO] - Epoch 17 with learning rate [0.0001] | |
| [2024-01-28 00:10:10,323][__main__][INFO] - loss : 0.14615798941502997 | |
| [2024-01-28 00:10:37,835][__main__][INFO] - Epoch 18 with learning rate [0.0001] | |
| [2024-01-28 00:37:34,062][__main__][INFO] - loss : 0.14384513116627923 | |
| [2024-01-28 00:38:03,615][__main__][INFO] - Epoch 19 with learning rate [0.0001] | |
| [2024-01-28 01:20:33,815][__main__][INFO] - loss : 0.14138763200740037 | |
| [2024-01-28 01:21:50,613][__main__][INFO] - Epoch 20 with learning rate [0.0001] | |
| [2024-01-28 02:09:39,269][__main__][INFO] - loss : 0.13947315612186997 | |
| [2024-01-28 02:11:00,016][__main__][INFO] - Epoch 21 with learning rate [0.0001] | |
| [2024-01-28 02:59:01,657][__main__][INFO] - loss : 0.13772385247051708 | |
| [2024-01-28 03:00:20,895][__main__][INFO] - Epoch 22 with learning rate [0.0001] | |
| [2024-01-28 03:48:16,466][__main__][INFO] - loss : 0.13618720066547416 | |
| [2024-01-28 03:49:35,789][__main__][INFO] - Epoch 23 with learning rate [0.0001] | |
| [2024-01-28 04:37:29,614][__main__][INFO] - loss : 0.13420913152893368 | |
| [2024-01-28 04:38:53,070][__main__][INFO] - Epoch 24 with learning rate [0.0001] | |
| [2024-01-28 05:26:58,011][__main__][INFO] - loss : 0.13257588320473815 | |
| [2024-01-28 05:28:20,849][__main__][INFO] - Train time [hour] : 13.391367570559185 h | |
| [2024-01-28 05:28:20,850][__main__][INFO] - Results saved in /home/bezzam/LenslessPiCam/outputs/2024-01-27/16-04-47 | |