[2024-01-28 01:35:14,289][__main__][INFO] - Using GPU for training.
[2024-01-28 01:35:16,586][__main__][INFO] - Reconstruction a few images with ADMM...
[2024-01-28 01:35:18,040][__main__][INFO] - Train test size : 24000
[2024-01-28 01:35:18,041][__main__][INFO] - Test test size : 999
[2024-01-28 01:35:18,684][__main__][INFO] - Training model with 34021330 parameters
[2024-01-28 01:35:18,685][__main__][INFO] - Setup time : 0.6440122127532959 s
[2024-01-28 01:35:18,685][__main__][INFO] - PSF shape : torch.Size([1, 135, 240, 3])
[2024-01-28 01:35:18,685][__main__][INFO] - Results saved in /home/bezzam/LenslessPiCam/outputs/2024-01-28/01-35-14
[2024-01-28 01:36:48,070][__main__][INFO] - Epoch 0 with learning rate [0.0001]
[2024-01-28 02:26:33,965][__main__][INFO] - loss : 0.7132078661719973
[2024-01-28 02:27:54,582][__main__][INFO] - Epoch 1 with learning rate [0.0001]
[2024-01-28 03:17:39,564][__main__][INFO] - loss : 0.623118721405664
[2024-01-28 03:19:04,805][__main__][INFO] - Epoch 2 with learning rate [0.0001]
[2024-01-28 04:08:51,781][__main__][INFO] - loss : 0.5947420155902697
[2024-01-28 04:10:14,803][__main__][INFO] - Epoch 3 with learning rate [0.0001]
[2024-01-28 05:00:26,731][__main__][INFO] - loss : 0.5760630624393617
[2024-01-28 05:01:49,294][__main__][INFO] - Epoch 4 with learning rate [0.0001]
[2024-01-28 05:41:26,904][__main__][INFO] - loss : 0.5637468313574783
[2024-01-28 05:41:58,292][__main__][INFO] - Epoch 5 with learning rate [0.0001]
[2024-01-28 06:09:59,060][__main__][INFO] - loss : 0.5536639285286256
[2024-01-28 06:10:30,642][__main__][INFO] - Epoch 6 with learning rate [0.0001]
[2024-01-28 06:38:29,496][__main__][INFO] - loss : 0.5443675012985875
[2024-01-28 06:39:02,002][__main__][INFO] - Epoch 7 with learning rate [0.0001]
[2024-01-28 07:07:01,868][__main__][INFO] - loss : 0.5373033655484504
[2024-01-28 07:07:33,786][__main__][INFO] - Epoch 8 with learning rate [0.0001]
[2024-01-28 07:35:32,507][__main__][INFO] - loss : 0.5305690682927763
[2024-01-28 07:36:03,473][__main__][INFO] - Epoch 9 with learning rate [0.0001]
[2024-01-28 08:04:03,312][__main__][INFO] - loss : 0.5242447232802697
[2024-01-28 08:04:34,056][__main__][INFO] - Epoch 10 with learning rate [0.0001]
[2024-01-28 08:32:33,604][__main__][INFO] - loss : 0.5177197459538778
[2024-01-28 08:33:04,297][__main__][INFO] - Epoch 11 with learning rate [0.0001]
[2024-01-28 09:01:04,514][__main__][INFO] - loss : 0.5124760565658406
[2024-01-28 09:01:35,176][__main__][INFO] - Epoch 12 with learning rate [0.0001]
[2024-01-28 09:36:35,706][__main__][INFO] - loss : 0.5073304253121202
[2024-01-28 09:37:56,656][__main__][INFO] - Epoch 13 with learning rate [0.0001]
[2024-01-28 10:28:48,288][__main__][INFO] - loss : 0.5032771647771189
[2024-01-28 10:30:12,349][__main__][INFO] - Epoch 14 with learning rate [0.0001]
[2024-01-28 11:20:45,695][__main__][INFO] - loss : 0.49700715555747343
[2024-01-28 11:22:08,406][__main__][INFO] - Epoch 15 with learning rate [0.0001]
[2024-01-28 12:12:36,689][__main__][INFO] - loss : 0.4927304321825499
[2024-01-28 12:13:58,846][__main__][INFO] - Epoch 16 with learning rate [0.0001]
[2024-01-28 13:03:58,606][__main__][INFO] - loss : 0.48797426237662594
[2024-01-28 13:05:22,935][__main__][INFO] - Epoch 17 with learning rate [0.0001]
[2024-01-28 13:55:55,163][__main__][INFO] - loss : 0.4847696518500643
[2024-01-28 13:57:17,247][__main__][INFO] - Epoch 18 with learning rate [0.0001]
[2024-01-28 14:48:07,660][__main__][INFO] - loss : 0.47971916896104816
[2024-01-28 14:49:28,553][__main__][INFO] - Epoch 19 with learning rate [0.0001]
[2024-01-28 15:40:09,139][__main__][INFO] - loss : 0.476704137951134
[2024-01-28 15:41:31,072][__main__][INFO] - Epoch 20 with learning rate [0.0001]
[2024-01-28 16:32:27,664][__main__][INFO] - loss : 0.47222046001752294
[2024-01-28 16:33:52,540][__main__][INFO] - Epoch 21 with learning rate [0.0001]
[2024-01-28 17:24:02,688][__main__][INFO] - loss : 0.46943070819974
[2024-01-28 17:25:27,098][__main__][INFO] - Epoch 22 with learning rate [0.0001]
[2024-01-28 18:15:49,366][__main__][INFO] - loss : 0.4654631819128995
[2024-01-28 18:17:12,767][__main__][INFO] - Epoch 23 with learning rate [0.0001]
[2024-01-28 19:08:01,234][__main__][INFO] - loss : 0.46218248893817215
[2024-01-28 19:09:23,746][__main__][INFO] - Epoch 24 with learning rate [0.0001]
[2024-01-28 20:00:15,623][__main__][INFO] - loss : 0.4590208643774196
[2024-01-28 20:01:36,642][__main__][INFO] - Train time [hour] : 18.43797159406874 h
[2024-01-28 20:01:36,643][__main__][INFO] - Results saved in /home/bezzam/LenslessPiCam/outputs/2024-01-28/01-35-14