[2024-01-29 20:08:48,585][__main__][INFO] - Using GPU for training.
[2024-01-29 20:08:49,729][__main__][INFO] - Reconstruction a few images with ADMM...
[2024-01-29 20:08:50,577][__main__][INFO] - Train test size : 24000
[2024-01-29 20:08:50,577][__main__][INFO] - Test test size : 999
[2024-01-29 20:08:51,023][__main__][INFO] - Training model with 34021330 parameters
[2024-01-29 20:08:51,023][__main__][INFO] - Setup time : 0.4457569122314453 s
[2024-01-29 20:08:51,023][__main__][INFO] - PSF shape : torch.Size([1, 135, 240, 3])
[2024-01-29 20:08:51,023][__main__][INFO] - Results saved in /home/bezzam/LenslessPiCam/outputs/2024-01-29/20-08-48
[2024-01-29 20:09:20,459][__main__][INFO] - Epoch 0 with learning rate [0.0001]
[2024-01-29 20:36:26,673][__main__][INFO] - loss : 0.31272996233403666
[2024-01-29 20:36:54,414][__main__][INFO] - Epoch 1 with learning rate [0.0001]
[2024-01-29 21:03:59,765][__main__][INFO] - loss : 0.2476788473675647
[2024-01-29 21:04:27,389][__main__][INFO] - Epoch 2 with learning rate [0.0001]
[2024-01-29 21:31:32,660][__main__][INFO] - loss : 0.22845874570806818
[2024-01-29 21:32:00,298][__main__][INFO] - Epoch 3 with learning rate [0.0001]
[2024-01-29 21:59:04,542][__main__][INFO] - loss : 0.21743685066699941
[2024-01-29 21:59:32,198][__main__][INFO] - Epoch 4 with learning rate [0.0001]
[2024-01-29 22:26:36,270][__main__][INFO] - loss : 0.20913450723389762
[2024-01-29 22:27:03,927][__main__][INFO] - Epoch 5 with learning rate [0.0001]
[2024-01-29 22:54:08,758][__main__][INFO] - loss : 0.20204510008792095
[2024-01-29 22:54:36,326][__main__][INFO] - Epoch 6 with learning rate [0.0001]
[2024-01-29 23:21:40,902][__main__][INFO] - loss : 0.19673348573843605
[2024-01-29 23:22:08,453][__main__][INFO] - Epoch 7 with learning rate [0.0001]
[2024-01-29 23:49:12,117][__main__][INFO] - loss : 0.19266361020008657
[2024-01-29 23:49:39,629][__main__][INFO] - Epoch 8 with learning rate [0.0001]
[2024-01-30 00:16:43,044][__main__][INFO] - loss : 0.18888920007149373
[2024-01-30 00:17:10,730][__main__][INFO] - Epoch 9 with learning rate [0.0001]
[2024-01-30 00:44:15,597][__main__][INFO] - loss : 0.18499369311332675
[2024-01-30 00:44:43,269][__main__][INFO] - Epoch 10 with learning rate [0.0001]
[2024-01-30 01:12:10,593][__main__][INFO] - loss : 0.1814239970693982
[2024-01-30 01:12:38,481][__main__][INFO] - Epoch 11 with learning rate [0.0001]
[2024-01-30 01:39:44,243][__main__][INFO] - loss : 0.178825319265326
[2024-01-30 01:40:11,871][__main__][INFO] - Epoch 12 with learning rate [0.0001]
[2024-01-30 02:07:34,470][__main__][INFO] - loss : 0.17606780401617286
[2024-01-30 02:08:01,966][__main__][INFO] - Epoch 13 with learning rate [0.0001]
[2024-01-30 02:35:16,267][__main__][INFO] - loss : 0.1731607075482604
[2024-01-30 02:35:43,868][__main__][INFO] - Epoch 14 with learning rate [0.0001]
[2024-01-30 03:02:47,321][__main__][INFO] - loss : 0.17031185934444276
[2024-01-30 03:03:14,900][__main__][INFO] - Epoch 15 with learning rate [0.0001]
[2024-01-30 03:30:18,766][__main__][INFO] - loss : 0.16778948743889735
[2024-01-30 03:30:46,248][__main__][INFO] - Epoch 16 with learning rate [0.0001]
[2024-01-30 03:57:50,137][__main__][INFO] - loss : 0.16531555780271665
[2024-01-30 03:58:17,764][__main__][INFO] - Epoch 17 with learning rate [0.0001]
[2024-01-30 04:25:21,100][__main__][INFO] - loss : 0.1635572496677439
[2024-01-30 04:25:48,807][__main__][INFO] - Epoch 18 with learning rate [0.0001]
[2024-01-30 04:52:52,858][__main__][INFO] - loss : 0.16152025172611087
[2024-01-30 04:53:20,233][__main__][INFO] - Epoch 19 with learning rate [0.0001]
[2024-01-30 05:20:24,491][__main__][INFO] - loss : 0.1591637803638975
[2024-01-30 05:20:51,839][__main__][INFO] - Epoch 20 with learning rate [0.0001]
[2024-01-30 05:47:55,187][__main__][INFO] - loss : 0.15764906018227362
[2024-01-30 05:48:22,975][__main__][INFO] - Epoch 21 with learning rate [0.0001]
[2024-01-30 06:15:26,451][__main__][INFO] - loss : 0.15561049388100712
[2024-01-30 06:15:54,050][__main__][INFO] - Epoch 22 with learning rate [0.0001]
[2024-01-30 06:42:57,149][__main__][INFO] - loss : 0.15436255330344056
[2024-01-30 06:43:24,644][__main__][INFO] - Epoch 23 with learning rate [0.0001]
[2024-01-30 07:10:28,037][__main__][INFO] - loss : 0.15234802578141265
[2024-01-30 07:10:55,736][__main__][INFO] - Epoch 24 with learning rate [0.0001]
[2024-01-30 07:37:59,540][__main__][INFO] - loss : 0.1506220602716005
[2024-01-30 07:38:27,060][__main__][INFO] - Train time [hour] : 11.493119131326676 h
[2024-01-30 07:38:27,060][__main__][INFO] - Results saved in /home/bezzam/LenslessPiCam/outputs/2024-01-29/20-08-48