| [2023-10-25 10:07:42,242][__main__][INFO] - Using GPU for training. | |
| [2023-10-25 10:07:44,459][__main__][INFO] - Data shape : torch.Size([1, 380, 507, 3]) | |
| [2023-10-25 10:07:44,460][__main__][INFO] - PSF shape : torch.Size([1, 380, 507, 3]) | |
| [2023-10-25 10:07:44,462][__main__][INFO] - PSF min : 0.0 | |
| [2023-10-25 10:07:44,464][__main__][INFO] - PSF max : 0.007229802664369345 | |
| [2023-10-25 10:07:44,464][__main__][INFO] - PSF dtype : torch.float32 | |
| [2023-10-25 10:07:44,466][__main__][INFO] - PSF norm : 1.0000030994415283 | |
| [2023-10-25 10:07:44,466][__main__][INFO] - Reconstruction a few images with ADMM... | |
| [2023-10-25 10:07:44,778][__main__][INFO] - Cropped shape : (262, 215, 3) | |
| [2023-10-25 10:07:46,330][__main__][INFO] - Train test size : 22100 | |
| [2023-10-25 10:07:46,330][__main__][INFO] - Test test size : 3900 | |
| [2023-10-25 10:07:46,344][__main__][INFO] - Training model with 40 parameters | |
| [2023-10-25 10:07:46,344][__main__][INFO] - Setup time : 0.014037609100341797 s | |
| [2023-10-25 10:07:46,344][__main__][INFO] - PSF shape : torch.Size([1, 380, 507, 3]) | |
| [2023-10-25 10:07:46,344][__main__][INFO] - Results saved in /home/bezzam/LenslessPiCam/outputs/2023-10-25/10-07-42 | |
| [2023-10-25 10:14:41,389][__main__][INFO] - Epoch 0 with learning rate [0.0001] | |
| [2023-10-25 11:13:55,920][__main__][INFO] - loss : 0.8802030037107512 | |
| [2023-10-25 11:18:06,645][__main__][INFO] - Epoch 1 with learning rate [0.0001] | |
| [2023-10-25 12:07:23,330][__main__][INFO] - loss : 0.8615266865626712 | |
| [2023-10-25 12:11:34,720][__main__][INFO] - Epoch 2 with learning rate [0.0001] | |
| [2023-10-25 13:00:50,814][__main__][INFO] - loss : 0.8441812430787426 | |
| [2023-10-25 13:05:01,783][__main__][INFO] - Epoch 3 with learning rate [0.0001] | |
| [2023-10-25 13:54:11,780][__main__][INFO] - loss : 0.8378802298636464 | |
| [2023-10-25 13:58:23,199][__main__][INFO] - Epoch 4 with learning rate [0.0001] | |
| [2023-10-25 14:47:38,505][__main__][INFO] - loss : 0.8351269757046431 | |
| [2023-10-25 14:51:50,585][__main__][INFO] - Epoch 5 with learning rate [0.0001] | |
| [2023-10-25 15:41:07,990][__main__][INFO] - loss : 0.832474038223334 | |
| [2023-10-25 15:45:18,863][__main__][INFO] - Epoch 6 with learning rate [0.0001] | |
| [2023-10-25 16:34:49,079][__main__][INFO] - loss : 0.8316764284582701 | |
| [2023-10-25 16:39:00,073][__main__][INFO] - Epoch 7 with learning rate [0.0001] | |
| [2023-10-25 17:28:18,952][__main__][INFO] - loss : 0.8300894484692573 | |
| [2023-10-25 17:32:28,584][__main__][INFO] - Epoch 8 with learning rate [0.0001] | |
| [2023-10-25 18:21:44,617][__main__][INFO] - loss : 0.8262994860955526 | |
| [2023-10-25 18:25:55,205][__main__][INFO] - Epoch 9 with learning rate [0.0001] | |
| [2023-10-25 19:15:01,193][__main__][INFO] - loss : 0.8244500280613268 | |
| [2023-10-25 19:19:12,029][__main__][INFO] - Epoch 10 with learning rate [0.0001] | |
| [2023-10-25 20:08:20,131][__main__][INFO] - loss : 0.8238364160654225 | |
| [2023-10-25 20:12:30,680][__main__][INFO] - Epoch 11 with learning rate [0.0001] | |
| [2023-10-25 21:01:37,280][__main__][INFO] - loss : 0.8232186751344045 | |
| [2023-10-25 21:05:46,826][__main__][INFO] - Epoch 12 with learning rate [0.0001] | |
| [2023-10-25 21:54:54,113][__main__][INFO] - loss : 0.82260233373124 | |
| [2023-10-25 21:59:04,138][__main__][INFO] - Epoch 13 with learning rate [0.0001] | |
| [2023-10-25 22:48:10,219][__main__][INFO] - loss : 0.8222699380460387 | |
| [2023-10-25 22:52:21,770][__main__][INFO] - Epoch 14 with learning rate [0.0001] | |
| [2023-10-25 23:41:28,812][__main__][INFO] - loss : 0.8218432146499584 | |
| [2023-10-25 23:45:38,977][__main__][INFO] - Epoch 15 with learning rate [0.0001] | |
| [2023-10-26 00:34:47,671][__main__][INFO] - loss : 0.8213285370110394 | |
| [2023-10-26 00:38:57,913][__main__][INFO] - Epoch 16 with learning rate [0.0001] | |
| [2023-10-26 01:28:05,334][__main__][INFO] - loss : 0.8211768602138194 | |
| [2023-10-26 01:32:14,850][__main__][INFO] - Epoch 17 with learning rate [0.0001] | |
| [2023-10-26 02:21:22,453][__main__][INFO] - loss : 0.8209193825074453 | |
| [2023-10-26 02:25:32,569][__main__][INFO] - Epoch 18 with learning rate [0.0001] | |
| [2023-10-26 03:14:40,254][__main__][INFO] - loss : 0.8204799105139328 | |
| [2023-10-26 03:18:50,479][__main__][INFO] - Epoch 19 with learning rate [0.0001] | |
| [2023-10-26 04:07:58,590][__main__][INFO] - loss : 0.8207494539373058 | |
| [2023-10-26 04:12:08,357][__main__][INFO] - Epoch 20 with learning rate [0.0001] | |
| [2023-10-26 05:01:14,891][__main__][INFO] - loss : 0.8192537037603451 | |
| [2023-10-26 05:05:24,450][__main__][INFO] - Epoch 21 with learning rate [0.0001] | |
| [2023-10-26 05:54:31,588][__main__][INFO] - loss : 0.8173046536359313 | |
| [2023-10-26 05:58:41,970][__main__][INFO] - Epoch 22 with learning rate [0.0001] | |
| [2023-10-26 06:47:50,976][__main__][INFO] - loss : 0.8157110446179083 | |
| [2023-10-26 06:52:00,448][__main__][INFO] - Epoch 23 with learning rate [0.0001] | |
| [2023-10-26 07:41:05,243][__main__][INFO] - loss : 0.81546365513521 | |
| [2023-10-26 07:45:16,216][__main__][INFO] - Epoch 24 with learning rate [0.0001] | |
| [2023-10-26 08:34:22,984][__main__][INFO] - loss : 0.8151451322827434 | |
| [2023-10-26 08:38:34,951][__main__][INFO] - Epoch 25 with learning rate [0.0001] | |