[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