[2024-01-16 10:16:56,537][__main__][INFO] - Using GPU for training.
[2024-01-16 10:16:58,471][__main__][INFO] - Reconstruction a few images with ADMM...
[2024-01-16 10:16:59,809][__main__][INFO] - Train test size : 24000
[2024-01-16 10:16:59,809][__main__][INFO] - Test test size : 999
[2024-01-16 10:16:59,822][__main__][INFO] - Training model with 80 parameters
[2024-01-16 10:16:59,822][__main__][INFO] - Setup time : 0.012391805648803711 s
[2024-01-16 10:16:59,822][__main__][INFO] - PSF shape : torch.Size([1, 135, 240, 3])
[2024-01-16 10:16:59,822][__main__][INFO] - Results saved in /home/bezzam/LenslessPiCam/outputs/2024-01-16/10-16-56
[2024-01-16 10:17:57,538][__main__][INFO] - Epoch 0 with learning rate [0.0001]
[2024-01-16 10:55:12,985][__main__][INFO] - loss : 0.6116351423660911
[2024-01-16 10:56:07,924][__main__][INFO] - Epoch 1 with learning rate [0.0001]
[2024-01-16 11:33:48,621][__main__][INFO] - loss : 0.5969407624204935
[2024-01-16 11:34:47,236][__main__][INFO] - Epoch 2 with learning rate [0.0001]
[2024-01-16 12:12:42,066][__main__][INFO] - loss : 0.5941597239673115
[2024-01-16 12:13:34,281][__main__][INFO] - Epoch 3 with learning rate [0.0001]
[2024-01-16 12:50:33,406][__main__][INFO] - loss : 0.5922832410633564
[2024-01-16 12:51:24,225][__main__][INFO] - Epoch 4 with learning rate [0.0001]
[2024-01-16 13:28:40,272][__main__][INFO] - loss : 0.5911119637191301
[2024-01-16 13:29:31,935][__main__][INFO] - Epoch 5 with learning rate [0.0001]
[2024-01-16 14:06:27,156][__main__][INFO] - loss : 0.5903311789333813
[2024-01-16 14:07:19,835][__main__][INFO] - Epoch 6 with learning rate [0.0001]
[2024-01-16 14:44:34,103][__main__][INFO] - loss : 0.5879827742079892
[2024-01-16 14:45:26,542][__main__][INFO] - Epoch 7 with learning rate [0.0001]
[2024-01-16 15:22:07,389][__main__][INFO] - loss : 0.5880078231096272
[2024-01-16 15:22:59,691][__main__][INFO] - Epoch 8 with learning rate [0.0001]
[2024-01-16 16:00:23,120][__main__][INFO] - loss : 0.586589498321215
[2024-01-16 16:01:18,359][__main__][INFO] - Epoch 9 with learning rate [0.0001]
[2024-01-16 16:40:20,823][__main__][INFO] - loss : 0.5863682059446956
[2024-01-16 16:41:22,714][__main__][INFO] - Epoch 10 with learning rate [0.0001]
[2024-01-16 17:19:29,997][__main__][INFO] - loss : 0.5857811225752035
[2024-01-16 17:20:25,079][__main__][INFO] - Epoch 11 with learning rate [0.0001]
[2024-01-16 17:57:27,285][__main__][INFO] - loss : 0.5851873279809958
[2024-01-16 17:58:21,145][__main__][INFO] - Epoch 12 with learning rate [0.0001]
[2024-01-16 18:35:37,230][__main__][INFO] - loss : 0.5853507430652778
[2024-01-16 18:36:30,836][__main__][INFO] - Epoch 13 with learning rate [0.0001]
[2024-01-16 19:02:12,390][__main__][INFO] - loss : 0.585124227970841
[2024-01-16 19:02:38,702][__main__][INFO] - Epoch 14 with learning rate [0.0001]
[2024-01-16 19:22:18,267][__main__][INFO] - loss : 0.5845710966388374
[2024-01-16 19:22:46,346][__main__][INFO] - Epoch 15 with learning rate [0.0001]
[2024-01-16 19:40:59,890][__main__][INFO] - loss : 0.5844923431177926
[2024-01-16 19:41:23,219][__main__][INFO] - Epoch 16 with learning rate [0.0001]
[2024-01-16 19:59:21,542][__main__][INFO] - loss : 0.5843840137918781
[2024-01-16 19:59:45,237][__main__][INFO] - Epoch 17 with learning rate [0.0001]
[2024-01-16 20:17:41,465][__main__][INFO] - loss : 0.5840132731497293
[2024-01-16 20:18:04,824][__main__][INFO] - Epoch 18 with learning rate [0.0001]
[2024-01-16 20:36:05,705][__main__][INFO] - loss : 0.5839742677410459
[2024-01-16 20:36:29,299][__main__][INFO] - Epoch 19 with learning rate [0.0001]
[2024-01-16 20:54:26,840][__main__][INFO] - loss : 0.5837308169702701
[2024-01-16 20:54:49,692][__main__][INFO] - Epoch 20 with learning rate [0.0001]
[2024-01-16 21:13:08,861][__main__][INFO] - loss : 0.5836687145431839
[2024-01-16 21:13:34,588][__main__][INFO] - Epoch 21 with learning rate [0.0001]
[2024-01-16 21:51:14,849][__main__][INFO] - loss : 0.5837951561311892
[2024-01-16 21:52:17,710][__main__][INFO] - Epoch 22 with learning rate [0.0001]
[2024-01-16 22:34:52,994][__main__][INFO] - loss : 0.5833144933879394
[2024-01-16 22:35:53,213][__main__][INFO] - Epoch 23 with learning rate [0.0001]
[2024-01-16 23:18:37,604][__main__][INFO] - loss : 0.5832784005800891
[2024-01-16 23:19:35,641][__main__][INFO] - Epoch 24 with learning rate [0.0001]
[2024-01-17 00:02:19,133][__main__][INFO] - loss : 0.5835274580419064
[2024-01-17 00:03:18,051][__main__][INFO] - Train time [hour] : 13.771451858348316 h
[2024-01-17 00:03:18,051][__main__][INFO] - Results saved in /home/bezzam/LenslessPiCam/outputs/2024-01-16/10-16-56