[2024-01-15 18:17:48,485][__main__][INFO] - Using GPU for training.
[2024-01-15 18:17:50,711][__main__][INFO] - Reconstruction a few images with ADMM...
[2024-01-15 18:17:52,019][__main__][INFO] - Train test size : 24000
[2024-01-15 18:17:52,019][__main__][INFO] - Test test size : 999
[2024-01-15 18:17:52,079][__main__][INFO] - Training model with 80 parameters
[2024-01-15 18:17:52,080][__main__][INFO] - Setup time : 0.06038069725036621 s
[2024-01-15 18:17:52,080][__main__][INFO] - PSF shape : torch.Size([1, 135, 240, 3])
[2024-01-15 18:17:52,080][__main__][INFO] - Results saved in /home/bezzam/LenslessPiCam/outputs/2024-01-15/18-17-48
[2024-01-15 18:18:41,903][__main__][INFO] - Epoch 0 with learning rate [0.0001]
[2024-01-15 18:55:49,535][__main__][INFO] - loss : 0.5186186236639828
[2024-01-15 18:56:34,741][__main__][INFO] - Epoch 1 with learning rate [0.0001]
[2024-01-15 19:33:49,106][__main__][INFO] - loss : 0.5080442267556975
[2024-01-15 19:34:38,063][__main__][INFO] - Epoch 2 with learning rate [0.0001]
[2024-01-15 20:11:34,968][__main__][INFO] - loss : 0.5058764455616469
[2024-01-15 20:12:21,668][__main__][INFO] - Epoch 3 with learning rate [0.0001]
[2024-01-15 20:49:27,046][__main__][INFO] - loss : 0.4928320770263682
[2024-01-15 20:50:11,849][__main__][INFO] - Epoch 4 with learning rate [0.0001]
[2024-01-15 21:27:08,618][__main__][INFO] - loss : 0.4915155341128515
[2024-01-15 21:27:53,247][__main__][INFO] - Epoch 5 with learning rate [0.0001]
[2024-01-15 22:05:02,075][__main__][INFO] - loss : 0.4906717370152475
[2024-01-15 22:05:55,072][__main__][INFO] - Epoch 6 with learning rate [0.0001]
[2024-01-15 22:43:03,717][__main__][INFO] - loss : 0.48995360862215276
[2024-01-15 22:43:48,385][__main__][INFO] - Epoch 7 with learning rate [0.0001]
[2024-01-15 23:20:52,568][__main__][INFO] - loss : 0.48953393385807725
[2024-01-15 23:21:41,114][__main__][INFO] - Epoch 8 with learning rate [0.0001]
[2024-01-15 23:58:55,710][__main__][INFO] - loss : 0.4892329586048927
[2024-01-15 23:59:41,028][__main__][INFO] - Epoch 9 with learning rate [0.0001]
[2024-01-16 00:36:48,636][__main__][INFO] - loss : 0.48896600639820154
[2024-01-16 00:37:32,274][__main__][INFO] - Epoch 10 with learning rate [0.0001]
[2024-01-16 01:14:40,814][__main__][INFO] - loss : 0.4887700859109561
[2024-01-16 01:15:26,447][__main__][INFO] - Epoch 11 with learning rate [0.0001]
[2024-01-16 01:52:23,913][__main__][INFO] - loss : 0.4884270545740922
[2024-01-16 01:53:08,164][__main__][INFO] - Epoch 12 with learning rate [0.0001]
[2024-01-16 02:30:17,003][__main__][INFO] - loss : 0.4883370875815552
[2024-01-16 02:31:05,185][__main__][INFO] - Epoch 13 with learning rate [0.0001]
[2024-01-16 03:08:21,089][__main__][INFO] - loss : 0.4881371052463846
[2024-01-16 03:09:06,065][__main__][INFO] - Epoch 14 with learning rate [0.0001]
[2024-01-16 03:43:00,721][__main__][INFO] - loss : 0.4881270435353114
[2024-01-16 03:43:38,831][__main__][INFO] - Epoch 15 with learning rate [0.0001]
[2024-01-16 04:16:05,266][__main__][INFO] - loss : 0.4879757748941581
[2024-01-16 04:16:43,701][__main__][INFO] - Epoch 16 with learning rate [0.0001]
[2024-01-16 04:49:17,550][__main__][INFO] - loss : 0.4877036250829695
[2024-01-16 04:49:55,819][__main__][INFO] - Epoch 17 with learning rate [0.0001]
[2024-01-16 05:22:20,800][__main__][INFO] - loss : 0.48768936632076865
[2024-01-16 05:22:59,054][__main__][INFO] - Epoch 18 with learning rate [0.0001]
[2024-01-16 05:55:30,808][__main__][INFO] - loss : 0.4876382187902927
[2024-01-16 05:56:08,840][__main__][INFO] - Epoch 19 with learning rate [0.0001]
[2024-01-16 06:28:34,127][__main__][INFO] - loss : 0.4872200195193287
[2024-01-16 06:29:12,730][__main__][INFO] - Epoch 20 with learning rate [0.0001]
[2024-01-16 07:01:45,859][__main__][INFO] - loss : 0.48694067114591555
[2024-01-16 07:02:24,209][__main__][INFO] - Epoch 21 with learning rate [0.0001]
[2024-01-16 07:34:51,005][__main__][INFO] - loss : 0.4867599181532862
[2024-01-16 07:35:28,865][__main__][INFO] - Epoch 22 with learning rate [0.0001]
[2024-01-16 08:07:59,799][__main__][INFO] - loss : 0.4868178767065206
[2024-01-16 08:08:38,440][__main__][INFO] - Epoch 23 with learning rate [0.0001]
[2024-01-16 08:41:36,769][__main__][INFO] - loss : 0.4864478222131731
[2024-01-16 08:42:22,987][__main__][INFO] - Epoch 24 with learning rate [0.0001]
[2024-01-16 09:18:12,819][__main__][INFO] - loss : 0.48673302693168274
[2024-01-16 09:18:57,626][__main__][INFO] - Train time [hour] : 15.017891580727365 h
[2024-01-16 09:18:57,626][__main__][INFO] - Results saved in /home/bezzam/LenslessPiCam/outputs/2024-01-15/18-17-48