[2024-01-18 18:15:02,612][__main__][INFO] - Using GPU for training.
[2024-01-18 18:15:04,221][__main__][INFO] - Reconstruction a few images with ADMM...
[2024-01-18 18:15:05,620][__main__][INFO] - Train test size : 24000
[2024-01-18 18:15:05,621][__main__][INFO] - Test test size : 999
[2024-01-18 18:15:05,768][__main__][INFO] - Training model with 17010705 parameters
[2024-01-18 18:15:05,768][__main__][INFO] - Setup time : 0.1476752758026123 s
[2024-01-18 18:15:05,768][__main__][INFO] - PSF shape : torch.Size([1, 135, 240, 3])
[2024-01-18 18:15:05,769][__main__][INFO] - Results saved in /home/bezzam/LenslessPiCam/outputs/2024-01-18/18-15-02
[2024-01-18 18:15:46,720][__main__][INFO] - Epoch 0 with learning rate [0.0001]
[2024-01-18 18:46:36,036][__main__][INFO] - loss : 0.4188102260828016
[2024-01-18 18:47:15,281][__main__][INFO] - Epoch 1 with learning rate [0.0001]
[2024-01-18 19:18:32,662][__main__][INFO] - loss : 0.33516882719596236
[2024-01-18 19:19:12,659][__main__][INFO] - Epoch 2 with learning rate [0.0001]
[2024-01-18 19:49:02,434][__main__][INFO] - loss : 0.31745233876506534
[2024-01-18 19:49:41,435][__main__][INFO] - Epoch 3 with learning rate [0.0001]
[2024-01-18 20:18:58,290][__main__][INFO] - loss : 0.30849826081097204
[2024-01-18 20:19:36,812][__main__][INFO] - Epoch 4 with learning rate [0.0001]
[2024-01-18 20:49:02,326][__main__][INFO] - loss : 0.3031157548675942
[2024-01-18 20:49:40,986][__main__][INFO] - Epoch 5 with learning rate [0.0001]
[2024-01-18 21:19:32,374][__main__][INFO] - loss : 0.29817252306143566
[2024-01-18 21:20:11,012][__main__][INFO] - Epoch 6 with learning rate [0.0001]
[2024-01-18 21:49:38,282][__main__][INFO] - loss : 0.2945401227027177
[2024-01-18 21:50:17,064][__main__][INFO] - Epoch 7 with learning rate [0.0001]
[2024-01-18 22:19:36,443][__main__][INFO] - loss : 0.29250809829930513
[2024-01-18 22:20:15,807][__main__][INFO] - Epoch 8 with learning rate [0.0001]
[2024-01-18 22:49:44,607][__main__][INFO] - loss : 0.2898601322025056
[2024-01-18 22:50:23,443][__main__][INFO] - Epoch 9 with learning rate [0.0001]
[2024-01-18 23:19:40,359][__main__][INFO] - loss : 0.2879842390567067
[2024-01-18 23:20:19,612][__main__][INFO] - Epoch 10 with learning rate [0.0001]
[2024-01-18 23:48:52,356][__main__][INFO] - loss : 0.28570315137505586
[2024-01-18 23:49:21,851][__main__][INFO] - Epoch 11 with learning rate [0.0001]
[2024-01-19 00:12:35,564][__main__][INFO] - loss : 0.2838655687918257
[2024-01-19 00:13:05,661][__main__][INFO] - Epoch 12 with learning rate [0.0001]
[2024-01-19 00:36:05,358][__main__][INFO] - loss : 0.2831620060751832
[2024-01-19 00:36:35,379][__main__][INFO] - Epoch 13 with learning rate [0.0001]
[2024-01-19 00:59:17,582][__main__][INFO] - loss : 0.28123869933684714
[2024-01-19 00:59:48,244][__main__][INFO] - Epoch 14 with learning rate [0.0001]
[2024-01-19 01:22:29,644][__main__][INFO] - loss : 0.2797611504445471
[2024-01-19 01:22:59,619][__main__][INFO] - Epoch 15 with learning rate [0.0001]
[2024-01-19 01:45:45,036][__main__][INFO] - loss : 0.2789651690820859
[2024-01-19 01:46:14,965][__main__][INFO] - Epoch 16 with learning rate [0.0001]
[2024-01-19 02:08:58,434][__main__][INFO] - loss : 0.27776181094348446
[2024-01-19 02:09:28,412][__main__][INFO] - Epoch 17 with learning rate [0.0001]
[2024-01-19 02:32:07,904][__main__][INFO] - loss : 0.27633536891142507
[2024-01-19 02:32:34,299][__main__][INFO] - Epoch 18 with learning rate [0.0001]
[2024-01-19 02:55:06,241][__main__][INFO] - loss : 0.2752036299755176
[2024-01-19 02:55:32,291][__main__][INFO] - Epoch 19 with learning rate [0.0001]
[2024-01-19 03:18:04,675][__main__][INFO] - loss : 0.27444457547366635
[2024-01-19 03:18:31,092][__main__][INFO] - Epoch 20 with learning rate [0.0001]
[2024-01-19 03:41:02,763][__main__][INFO] - loss : 0.2737527764787267
[2024-01-19 03:41:29,082][__main__][INFO] - Epoch 21 with learning rate [0.0001]
[2024-01-19 04:04:00,990][__main__][INFO] - loss : 0.2727003255238143
[2024-01-19 04:04:27,056][__main__][INFO] - Epoch 22 with learning rate [0.0001]
[2024-01-19 04:26:58,140][__main__][INFO] - loss : 0.271493462348977
[2024-01-19 04:27:24,388][__main__][INFO] - Epoch 23 with learning rate [0.0001]
[2024-01-19 04:49:58,334][__main__][INFO] - loss : 0.27124925253291876
[2024-01-19 04:50:24,773][__main__][INFO] - Epoch 24 with learning rate [0.0001]
[2024-01-19 05:12:57,224][__main__][INFO] - loss : 0.2702767081558715
[2024-01-19 05:13:23,468][__main__][INFO] - Train time [hour] : 10.971332937346565 h
[2024-01-19 05:13:23,469][__main__][INFO] - Results saved in /home/bezzam/LenslessPiCam/outputs/2024-01-18/18-15-02