[2023-10-26 11:52:05,787][__main__][INFO] - Using GPU for training.
[2023-10-26 11:52:07,426][__main__][INFO] - Data shape :  torch.Size([1, 380, 507, 3])
[2023-10-26 11:52:07,427][__main__][INFO] - PSF shape : torch.Size([1, 380, 507, 3])
[2023-10-26 11:52:07,428][__main__][INFO] - PSF min : 0.0
[2023-10-26 11:52:07,428][__main__][INFO] - PSF max : 0.007229802664369345
[2023-10-26 11:52:07,428][__main__][INFO] - PSF dtype : torch.float32
[2023-10-26 11:52:07,429][__main__][INFO] - PSF norm : 1.0000030994415283
[2023-10-26 11:52:07,429][__main__][INFO] - Reconstruction a few images with ADMM...
[2023-10-26 11:52:07,643][__main__][INFO] - Cropped shape :  (262, 215, 3)
[2023-10-26 11:52:09,011][__main__][INFO] - Train test size : 22100
[2023-10-26 11:52:09,011][__main__][INFO] - Test test size : 3900
[2023-10-26 11:52:09,077][__main__][INFO] - Training model with 8161289 parameters
[2023-10-26 11:52:09,077][__main__][INFO] - Setup time : 0.06639313697814941 s
[2023-10-26 11:52:09,077][__main__][INFO] - PSF shape : torch.Size([1, 380, 507, 3])
[2023-10-26 11:52:09,077][__main__][INFO] - Results saved in /home/bezzam/LenslessPiCam/outputs/2023-10-26/11-52-05
[2023-10-26 11:57:11,432][__main__][INFO] - Epoch 0 with learning rate [0.0001]
[2023-10-26 13:04:27,536][__main__][INFO] - loss : 0.6876848807162274
[2023-10-26 13:09:26,707][__main__][INFO] - Epoch 1 with learning rate [0.0001]
[2023-10-26 14:16:41,795][__main__][INFO] - loss : 0.6284913022701553
[2023-10-26 14:21:41,027][__main__][INFO] - Epoch 2 with learning rate [0.0001]
[2023-10-26 15:28:56,037][__main__][INFO] - loss : 0.5986766646350681
[2023-10-26 15:33:56,333][__main__][INFO] - Epoch 3 with learning rate [0.0001]
[2023-10-26 16:41:08,143][__main__][INFO] - loss : 0.5157111108626723
[2023-10-26 16:46:08,937][__main__][INFO] - Epoch 4 with learning rate [0.0001]
[2023-10-26 17:53:44,541][__main__][INFO] - loss : 0.4958108805531283
[2023-10-26 17:58:43,265][__main__][INFO] - Epoch 5 with learning rate [0.0001]
[2023-10-26 19:06:18,606][__main__][INFO] - loss : 0.48538456920584994
[2023-10-26 19:11:17,474][__main__][INFO] - Epoch 6 with learning rate [0.0001]
[2023-10-26 20:18:30,084][__main__][INFO] - loss : 0.4780645218585957
[2023-10-26 20:23:29,029][__main__][INFO] - Epoch 7 with learning rate [0.0001]
[2023-10-26 21:30:40,410][__main__][INFO] - loss : 0.4724020235247202
[2023-10-26 21:35:38,517][__main__][INFO] - Epoch 8 with learning rate [0.0001]
[2023-10-26 22:42:48,151][__main__][INFO] - loss : 0.46755817437603614
[2023-10-26 22:47:47,350][__main__][INFO] - Epoch 9 with learning rate [0.0001]
[2023-10-26 23:54:59,648][__main__][INFO] - loss : 0.4637158002303191
[2023-10-26 23:59:58,855][__main__][INFO] - Epoch 10 with learning rate [0.0001]
[2023-10-27 01:07:09,015][__main__][INFO] - loss : 0.4602826083947087
[2023-10-27 01:12:07,574][__main__][INFO] - Epoch 11 with learning rate [0.0001]
[2023-10-27 02:19:16,466][__main__][INFO] - loss : 0.4569949800504271
[2023-10-27 02:24:14,823][__main__][INFO] - Epoch 12 with learning rate [0.0001]
[2023-10-27 03:31:22,177][__main__][INFO] - loss : 0.45435437462448564
[2023-10-27 03:36:20,721][__main__][INFO] - Epoch 13 with learning rate [0.0001]
[2023-10-27 04:43:28,551][__main__][INFO] - loss : 0.4523512957214767
[2023-10-27 04:48:27,073][__main__][INFO] - Epoch 14 with learning rate [0.0001]
[2023-10-27 05:55:32,554][__main__][INFO] - loss : 0.45015358741466704
[2023-10-27 06:00:31,229][__main__][INFO] - Epoch 15 with learning rate [0.0001]
[2023-10-27 07:07:42,078][__main__][INFO] - loss : 0.4481310900681699
[2023-10-27 07:12:40,722][__main__][INFO] - Epoch 16 with learning rate [0.0001]
[2023-10-27 08:19:58,857][__main__][INFO] - loss : 0.4471155107722566
[2023-10-27 08:25:00,134][__main__][INFO] - Epoch 17 with learning rate [0.0001]
[2023-10-27 09:32:21,588][__main__][INFO] - loss : 0.4455031084798576
[2023-10-27 09:37:22,074][__main__][INFO] - Epoch 18 with learning rate [0.0001]
[2023-10-27 10:44:42,029][__main__][INFO] - loss : 0.44390673161631755
[2023-10-27 10:49:43,748][__main__][INFO] - Epoch 19 with learning rate [0.0001]
[2023-10-27 11:57:30,709][__main__][INFO] - loss : 0.442675518789982
[2023-10-27 12:02:30,771][__main__][INFO] - Epoch 20 with learning rate [0.0001]
[2023-10-27 13:09:51,546][__main__][INFO] - loss : 0.4413972915621363
[2023-10-27 13:14:52,113][__main__][INFO] - Epoch 21 with learning rate [0.0001]
[2023-10-27 14:22:08,643][__main__][INFO] - loss : 0.4406492271466495
[2023-10-27 14:27:09,429][__main__][INFO] - Epoch 22 with learning rate [0.0001]
[2023-10-27 15:34:26,801][__main__][INFO] - loss : 0.4393968736909626
[2023-10-27 15:39:28,322][__main__][INFO] - Epoch 23 with learning rate [0.0001]
[2023-10-27 16:46:48,270][__main__][INFO] - loss : 0.43855327457324433
[2023-10-27 16:51:53,258][__main__][INFO] - Epoch 24 with learning rate [0.0001]
[2023-10-27 17:59:12,178][__main__][INFO] - loss : 0.43763846977803483
[2023-10-27 18:04:13,547][__main__][INFO] - Epoch 25 with learning rate [0.0001]
[2023-10-27 19:11:30,428][__main__][INFO] - loss : 0.4362611042751985
[2023-10-27 19:16:31,297][__main__][INFO] - Epoch 26 with learning rate [0.0001]
[2023-10-27 20:23:47,110][__main__][INFO] - loss : 0.4353535123917859
[2023-10-27 20:28:47,813][__main__][INFO] - Epoch 27 with learning rate [0.0001]
[2023-10-27 21:36:06,505][__main__][INFO] - loss : 0.43542409903323426
[2023-10-27 21:41:06,498][__main__][INFO] - Epoch 28 with learning rate [0.0001]
[2023-10-27 22:48:19,961][__main__][INFO] - loss : 0.4342909221034239
[2023-10-27 22:53:20,502][__main__][INFO] - Epoch 29 with learning rate [0.0001]