[2024-01-25 17:43:29,458][__main__][INFO] - Using GPU for training.
[2024-01-25 17:43:31,679][__main__][INFO] - Reconstruction a few images with ADMM...
[2024-01-25 17:43:33,262][__main__][INFO] - Train test size : 24000
[2024-01-25 17:43:33,263][__main__][INFO] - Test test size : 999
[2024-01-25 17:43:33,767][__main__][INFO] - Training model with 32738161 parameters
[2024-01-25 17:43:33,768][__main__][INFO] - Setup time : 0.5050997734069824 s
[2024-01-25 17:43:33,768][__main__][INFO] - PSF shape : torch.Size([1, 135, 240, 3])
[2024-01-25 17:43:33,768][__main__][INFO] - Results saved in /home/bezzam/LenslessPiCam/outputs/2024-01-25/17-43-29
[2024-01-25 17:44:56,823][__main__][INFO] - Epoch 0 with learning rate [0.0001]
[2024-01-25 18:24:48,679][__main__][INFO] - loss : 0.42506172037124607
[2024-01-25 18:26:09,293][__main__][INFO] - Epoch 1 with learning rate [0.0001]
[2024-01-25 19:05:58,393][__main__][INFO] - loss : 0.3715379193723203
[2024-01-25 19:07:18,251][__main__][INFO] - Epoch 2 with learning rate [0.0001]
[2024-01-25 19:46:56,253][__main__][INFO] - loss : 0.3513330962657927
[2024-01-25 19:48:16,570][__main__][INFO] - Epoch 3 with learning rate [0.0001]
[2024-01-25 20:07:22,239][__main__][INFO] - loss : 0.33664788734912926
[2024-01-25 20:07:38,938][__main__][INFO] - Epoch 4 with learning rate [0.0001]
[2024-01-25 20:20:37,098][__main__][INFO] - loss : 0.32508560132483544
[2024-01-25 20:20:54,012][__main__][INFO] - Epoch 5 with learning rate [0.0001]
[2024-01-25 20:33:53,297][__main__][INFO] - loss : 0.31761638724307173
[2024-01-25 20:34:10,909][__main__][INFO] - Epoch 6 with learning rate [0.0001]
[2024-01-25 20:47:08,539][__main__][INFO] - loss : 0.3107886923700576
[2024-01-25 20:47:26,074][__main__][INFO] - Epoch 7 with learning rate [0.0001]
[2024-01-25 21:00:22,379][__main__][INFO] - loss : 0.3049010163992641
[2024-01-25 21:00:39,054][__main__][INFO] - Epoch 8 with learning rate [0.0001]
[2024-01-25 21:13:35,620][__main__][INFO] - loss : 0.29941335070629865
[2024-01-25 21:13:53,196][__main__][INFO] - Epoch 9 with learning rate [0.0001]
[2024-01-25 21:26:51,404][__main__][INFO] - loss : 0.2952556682278715
[2024-01-25 21:27:08,669][__main__][INFO] - Epoch 10 with learning rate [0.0001]
[2024-01-25 21:40:05,099][__main__][INFO] - loss : 0.2908715750028692
[2024-01-25 21:40:21,494][__main__][INFO] - Epoch 11 with learning rate [0.0001]
[2024-01-25 21:53:14,970][__main__][INFO] - loss : 0.2871121711333593
[2024-01-25 21:53:31,394][__main__][INFO] - Epoch 12 with learning rate [0.0001]
[2024-01-25 22:06:25,855][__main__][INFO] - loss : 0.2836354361226164
[2024-01-25 22:06:42,240][__main__][INFO] - Epoch 13 with learning rate [0.0001]
[2024-01-25 22:19:35,705][__main__][INFO] - loss : 0.27971938257416235
[2024-01-25 22:19:52,161][__main__][INFO] - Epoch 14 with learning rate [0.0001]
[2024-01-25 22:32:43,573][__main__][INFO] - loss : 0.2761586771557726
[2024-01-25 22:32:59,523][__main__][INFO] - Epoch 15 with learning rate [0.0001]
[2024-01-25 22:45:53,160][__main__][INFO] - loss : 0.27307987784842636
[2024-01-25 22:46:11,070][__main__][INFO] - Epoch 16 with learning rate [0.0001]
[2024-01-25 22:59:05,409][__main__][INFO] - loss : 0.269839598769943
[2024-01-25 22:59:23,087][__main__][INFO] - Epoch 17 with learning rate [0.0001]
[2024-01-25 23:12:17,260][__main__][INFO] - loss : 0.2671173327267169
[2024-01-25 23:12:34,912][__main__][INFO] - Epoch 18 with learning rate [0.0001]
[2024-01-25 23:25:26,143][__main__][INFO] - loss : 0.26434651042024365
[2024-01-25 23:25:43,128][__main__][INFO] - Epoch 19 with learning rate [0.0001]
[2024-01-25 23:38:36,653][__main__][INFO] - loss : 0.2615345354080202
[2024-01-25 23:38:54,382][__main__][INFO] - Epoch 20 with learning rate [0.0001]
[2024-01-25 23:51:46,856][__main__][INFO] - loss : 0.2592328493793806
[2024-01-25 23:52:03,763][__main__][INFO] - Epoch 21 with learning rate [0.0001]
[2024-01-26 00:04:57,997][__main__][INFO] - loss : 0.25651818836232126
[2024-01-26 00:05:14,157][__main__][INFO] - Epoch 22 with learning rate [0.0001]
[2024-01-26 00:18:08,774][__main__][INFO] - loss : 0.2541421487977099
[2024-01-26 00:18:25,562][__main__][INFO] - Epoch 23 with learning rate [0.0001]
[2024-01-26 00:31:21,350][__main__][INFO] - loss : 0.251509060263634
[2024-01-26 00:31:37,978][__main__][INFO] - Epoch 24 with learning rate [0.0001]
[2024-01-26 00:44:32,539][__main__][INFO] - loss : 0.24944406007726969
[2024-01-26 00:44:48,670][__main__][INFO] - Train time [hour] : 7.020398301680883 h
[2024-01-26 00:44:48,670][__main__][INFO] - Results saved in /home/bezzam/LenslessPiCam/outputs/2024-01-25/17-43-29