[2024-01-26 12:33:01,806][__main__][INFO] - Using GPU for training.
[2024-01-26 12:33:04,237][__main__][INFO] - Reconstruction a few images with ADMM...
[2024-01-26 12:33:05,990][__main__][INFO] - Train test size : 24000
[2024-01-26 12:33:05,990][__main__][INFO] - Test test size : 999
[2024-01-26 12:33:07,153][__main__][INFO] - Training model with 34118450 parameters
[2024-01-26 12:33:07,153][__main__][INFO] - Setup time : 1.1631758213043213 s
[2024-01-26 12:33:07,154][__main__][INFO] - PSF shape : torch.Size([1, 135, 240, 3])
[2024-01-26 12:33:07,154][__main__][INFO] - Results saved in /home/bezzam/LenslessPiCam/outputs/2024-01-26/12-33-01
[2024-01-26 12:34:31,321][__main__][INFO] - Epoch 0 with learning rate [0.0001]
[2024-01-26 13:14:06,867][__main__][INFO] - loss : 0.3248639020870128
[2024-01-26 13:15:28,953][__main__][INFO] - Epoch 1 with learning rate [0.0001]
[2024-01-26 13:55:06,724][__main__][INFO] - loss : 0.282832590823372
[2024-01-26 13:56:28,230][__main__][INFO] - Epoch 2 with learning rate [0.0001]
[2024-01-26 14:36:13,543][__main__][INFO] - loss : 0.27184186743199845
[2024-01-26 14:37:33,544][__main__][INFO] - Epoch 3 with learning rate [0.0001]
[2024-01-26 15:17:16,195][__main__][INFO] - loss : 0.2637478157679241
[2024-01-26 15:18:38,507][__main__][INFO] - Epoch 4 with learning rate [0.0001]
[2024-01-26 15:58:24,261][__main__][INFO] - loss : 0.25708124168713925
[2024-01-26 15:59:45,479][__main__][INFO] - Epoch 5 with learning rate [0.0001]
[2024-01-26 16:39:30,767][__main__][INFO] - loss : 0.2515942390759785
[2024-01-26 16:40:51,451][__main__][INFO] - Epoch 6 with learning rate [0.0001]
[2024-01-26 17:20:51,807][__main__][INFO] - loss : 0.24692488486568145
[2024-01-26 17:22:14,499][__main__][INFO] - Epoch 7 with learning rate [0.0001]
[2024-01-26 18:02:17,594][__main__][INFO] - loss : 0.2431115454634031
[2024-01-26 18:03:39,003][__main__][INFO] - Epoch 8 with learning rate [0.0001]
[2024-01-26 18:43:39,262][__main__][INFO] - loss : 0.23959121106564993
[2024-01-26 18:45:03,434][__main__][INFO] - Epoch 9 with learning rate [0.0001]
[2024-01-26 19:25:06,383][__main__][INFO] - loss : 0.2363328355600443
[2024-01-26 19:26:26,927][__main__][INFO] - Epoch 10 with learning rate [0.0001]
[2024-01-26 20:05:35,796][__main__][INFO] - loss : 0.23332378724714117
[2024-01-26 20:06:51,538][__main__][INFO] - Epoch 11 with learning rate [0.0001]
[2024-01-26 20:45:34,403][__main__][INFO] - loss : 0.2302502313703293
[2024-01-26 20:46:47,990][__main__][INFO] - Epoch 12 with learning rate [0.0001]
[2024-01-26 21:25:37,028][__main__][INFO] - loss : 0.227525670826435
[2024-01-26 21:26:51,317][__main__][INFO] - Epoch 13 with learning rate [0.0001]
[2024-01-26 22:05:47,835][__main__][INFO] - loss : 0.22469300311803828
[2024-01-26 22:07:03,199][__main__][INFO] - Epoch 14 with learning rate [0.0001]
[2024-01-26 22:45:42,895][__main__][INFO] - loss : 0.22213923559586232
[2024-01-26 22:46:57,545][__main__][INFO] - Epoch 15 with learning rate [0.0001]
[2024-01-26 23:25:35,884][__main__][INFO] - loss : 0.2196691601077715
[2024-01-26 23:26:47,464][__main__][INFO] - Epoch 16 with learning rate [0.0001]
[2024-01-27 00:05:19,860][__main__][INFO] - loss : 0.21728063804407932
[2024-01-27 00:06:30,104][__main__][INFO] - Epoch 17 with learning rate [0.0001]
[2024-01-27 00:45:15,963][__main__][INFO] - loss : 0.2149266833811995
[2024-01-27 00:46:31,959][__main__][INFO] - Epoch 18 with learning rate [0.0001]
[2024-01-27 01:25:17,730][__main__][INFO] - loss : 0.21248165821532383
[2024-01-27 01:26:36,710][__main__][INFO] - Epoch 19 with learning rate [0.0001]
[2024-01-27 02:05:12,394][__main__][INFO] - loss : 0.21020547036578255
[2024-01-27 02:06:30,991][__main__][INFO] - Epoch 20 with learning rate [0.0001]
[2024-01-27 02:45:18,077][__main__][INFO] - loss : 0.2080953121930357
[2024-01-27 02:46:39,051][__main__][INFO] - Epoch 21 with learning rate [0.0001]
[2024-01-27 03:25:16,807][__main__][INFO] - loss : 0.20600388763844968
[2024-01-27 03:26:37,065][__main__][INFO] - Epoch 22 with learning rate [0.0001]
[2024-01-27 04:05:13,867][__main__][INFO] - loss : 0.2041021430293722
[2024-01-27 04:05:59,490][__main__][INFO] - Epoch 23 with learning rate [0.0001]
[2024-01-27 04:19:32,297][__main__][INFO] - loss : 0.2021262625406187
[2024-01-27 04:19:49,873][__main__][INFO] - Epoch 24 with learning rate [0.0001]
[2024-01-27 04:33:37,082][__main__][INFO] - loss : 0.20028283561269475
[2024-01-27 04:34:22,344][__main__][INFO] - Train time [hour] : 16.02047075410684 h
[2024-01-27 04:34:22,345][__main__][INFO] - Results saved in /home/bezzam/LenslessPiCam/outputs/2024-01-26/12-33-01