[2023-10-29 12:47:05,247][__main__][INFO] - Using GPU for training.
[2023-10-29 12:47:07,224][__main__][INFO] - Data shape :  torch.Size([1, 380, 507, 3])
[2023-10-29 12:47:07,225][__main__][INFO] - PSF shape : torch.Size([1, 380, 507, 3])
[2023-10-29 12:47:07,226][__main__][INFO] - PSF min : 0.0
[2023-10-29 12:47:07,226][__main__][INFO] - PSF max : 0.026563314720988274
[2023-10-29 12:47:07,226][__main__][INFO] - PSF dtype : torch.float32
[2023-10-29 12:47:07,226][__main__][INFO] - PSF norm : 1.0
[2023-10-29 12:47:07,227][__main__][INFO] - Reconstruction a few images with ADMM...
[2023-10-29 12:47:07,442][__main__][INFO] - Cropped shape :  (262, 215, 3)
[2023-10-29 12:47:08,900][__main__][INFO] - Train test size : 22100
[2023-10-29 12:47:08,900][__main__][INFO] - Test test size : 3900
[2023-10-29 12:47:08,964][__main__][INFO] - Training model with 8161249 parameters
[2023-10-29 12:47:08,964][__main__][INFO] - Setup time : 0.06374621391296387 s
[2023-10-29 12:47:08,964][__main__][INFO] - PSF shape : torch.Size([1, 380, 507, 3])
[2023-10-29 12:47:08,964][__main__][INFO] - Results saved in /home/bezzam/LenslessPiCam/outputs/2023-10-29/12-47-05
[2023-10-29 12:50:06,958][__main__][INFO] - Epoch 0 with learning rate [0.0001]
[2023-10-29 13:21:24,414][__main__][INFO] - loss : 0.6644346823843352
[2023-10-29 13:24:17,572][__main__][INFO] - Epoch 1 with learning rate [0.0001]
[2023-10-29 13:55:30,476][__main__][INFO] - loss : 0.5399755331671625
[2023-10-29 13:58:23,273][__main__][INFO] - Epoch 2 with learning rate [0.0001]
[2023-10-29 14:29:38,061][__main__][INFO] - loss : 0.48706577743340373
[2023-10-29 14:32:29,736][__main__][INFO] - Epoch 3 with learning rate [0.0001]
[2023-10-29 15:03:43,088][__main__][INFO] - loss : 0.4641708921684944
[2023-10-29 15:06:36,415][__main__][INFO] - Epoch 4 with learning rate [0.0001]
[2023-10-29 15:37:48,895][__main__][INFO] - loss : 0.44987145758861863
[2023-10-29 15:40:42,828][__main__][INFO] - Epoch 5 with learning rate [0.0001]
[2023-10-29 16:11:49,442][__main__][INFO] - loss : 0.43843807580783944
[2023-10-29 16:14:43,038][__main__][INFO] - Epoch 6 with learning rate [0.0001]
[2023-10-29 16:45:52,494][__main__][INFO] - loss : 0.4301338604672463
[2023-10-29 16:48:46,296][__main__][INFO] - Epoch 7 with learning rate [0.0001]
[2023-10-29 17:19:58,171][__main__][INFO] - loss : 0.42332181540549646
[2023-10-29 17:22:52,150][__main__][INFO] - Epoch 8 with learning rate [0.0001]
[2023-10-29 17:54:00,490][__main__][INFO] - loss : 0.4173950698310967
[2023-10-29 17:56:54,090][__main__][INFO] - Epoch 9 with learning rate [0.0001]
[2023-10-29 18:28:04,589][__main__][INFO] - loss : 0.41263551129474935
[2023-10-29 18:30:58,445][__main__][INFO] - Epoch 10 with learning rate [0.0001]
[2023-10-29 19:02:06,547][__main__][INFO] - loss : 0.40820048477315213
[2023-10-29 19:04:59,723][__main__][INFO] - Epoch 11 with learning rate [0.0001]
[2023-10-29 19:36:07,544][__main__][INFO] - loss : 0.40436135740841295
[2023-10-29 19:39:00,582][__main__][INFO] - Epoch 12 with learning rate [0.0001]
[2023-10-29 20:10:12,007][__main__][INFO] - loss : 0.4011661840186394
[2023-10-29 20:13:05,624][__main__][INFO] - Epoch 13 with learning rate [0.0001]
[2023-10-29 20:44:14,166][__main__][INFO] - loss : 0.39799810959742704
[2023-10-29 20:47:07,807][__main__][INFO] - Epoch 14 with learning rate [0.0001]
[2023-10-29 21:18:17,766][__main__][INFO] - loss : 0.39506083137309356
[2023-10-29 21:21:11,329][__main__][INFO] - Epoch 15 with learning rate [0.0001]
[2023-10-29 21:52:20,234][__main__][INFO] - loss : 0.3924017314889318
[2023-10-29 21:55:14,126][__main__][INFO] - Epoch 16 with learning rate [0.0001]
[2023-10-29 22:26:24,286][__main__][INFO] - loss : 0.38964456820380083
[2023-10-29 22:29:18,114][__main__][INFO] - Epoch 17 with learning rate [0.0001]
[2023-10-29 23:00:28,256][__main__][INFO] - loss : 0.3876190448077023
[2023-10-29 23:03:21,705][__main__][INFO] - Epoch 18 with learning rate [0.0001]
[2023-10-29 23:34:35,107][__main__][INFO] - loss : 0.38534067812009176
[2023-10-29 23:37:28,931][__main__][INFO] - Epoch 19 with learning rate [0.0001]
[2023-10-30 00:08:39,935][__main__][INFO] - loss : 0.3832974826588353
[2023-10-30 00:11:33,665][__main__][INFO] - Epoch 20 with learning rate [0.0001]
[2023-10-30 00:42:44,654][__main__][INFO] - loss : 0.3809599129862376
[2023-10-30 00:45:38,415][__main__][INFO] - Epoch 21 with learning rate [0.0001]
[2023-10-30 01:16:46,906][__main__][INFO] - loss : 0.3795172518141133
[2023-10-30 01:19:40,250][__main__][INFO] - Epoch 22 with learning rate [0.0001]
[2023-10-30 01:50:49,466][__main__][INFO] - loss : 0.37748991531214654
[2023-10-30 01:53:43,757][__main__][INFO] - Epoch 23 with learning rate [0.0001]
[2023-10-30 02:24:58,040][__main__][INFO] - loss : 0.37625316615018345
[2023-10-30 02:27:52,885][__main__][INFO] - Epoch 24 with learning rate [0.0001]
[2023-10-30 02:59:02,299][__main__][INFO] - loss : 0.37449975396982677
[2023-10-30 03:01:56,189][__main__][INFO] - Train time : 51286.40317463875 s
[2023-10-30 03:01:56,189][__main__][INFO] - Results saved in /home/bezzam/LenslessPiCam/outputs/2023-10-29/12-47-05