| [2024-01-17 20:06:45,733][__main__][INFO] - Using GPU for training. | |
| [2024-01-17 20:06:47,198][__main__][INFO] - Reconstruction a few images with ADMM... | |
| [2024-01-17 20:06:48,262][__main__][INFO] - Train test size : 24000 | |
| [2024-01-17 20:06:48,262][__main__][INFO] - Test test size : 999 | |
| [2024-01-17 20:06:48,375][__main__][INFO] - Training model with 17010705 parameters | |
| [2024-01-17 20:06:48,376][__main__][INFO] - Setup time : 0.11329770088195801 s | |
| [2024-01-17 20:06:48,376][__main__][INFO] - PSF shape : torch.Size([1, 135, 240, 3]) | |
| [2024-01-17 20:06:48,376][__main__][INFO] - Results saved in /home/bezzam/LenslessPiCam/outputs/2024-01-17/20-06-45 | |
| [2024-01-17 20:07:23,069][__main__][INFO] - Epoch 0 with learning rate [0.0001] | |
| [2024-01-17 20:33:23,142][__main__][INFO] - loss : 0.36667297692100115 | |
| [2024-01-17 20:33:55,814][__main__][INFO] - Epoch 1 with learning rate [0.0001] | |
| [2024-01-17 20:59:54,554][__main__][INFO] - loss : 0.2878954428384699 | |
| [2024-01-17 21:00:27,578][__main__][INFO] - Epoch 2 with learning rate [0.0001] | |
| [2024-01-17 21:26:08,478][__main__][INFO] - loss : 0.26462542621791285 | |
| [2024-01-17 21:26:40,818][__main__][INFO] - Epoch 3 with learning rate [0.0001] | |
| [2024-01-17 21:52:25,994][__main__][INFO] - loss : 0.2509993498772382 | |
| [2024-01-17 21:52:58,753][__main__][INFO] - Epoch 4 with learning rate [0.0001] | |
| [2024-01-17 22:18:53,004][__main__][INFO] - loss : 0.24276707645257278 | |
| [2024-01-17 22:19:26,145][__main__][INFO] - Epoch 5 with learning rate [0.0001] | |
| [2024-01-17 22:45:09,006][__main__][INFO] - loss : 0.23589916938543318 | |
| [2024-01-17 22:45:41,630][__main__][INFO] - Epoch 6 with learning rate [0.0001] | |
| [2024-01-17 23:11:29,911][__main__][INFO] - loss : 0.23032548744976525 | |
| [2024-01-17 23:12:03,028][__main__][INFO] - Epoch 7 with learning rate [0.0001] | |
| [2024-01-17 23:37:46,201][__main__][INFO] - loss : 0.2260298660149175 | |
| [2024-01-17 23:38:16,925][__main__][INFO] - Epoch 8 with learning rate [0.0001] | |
| [2024-01-18 00:04:13,568][__main__][INFO] - loss : 0.2212433759719131 | |
| [2024-01-18 00:04:47,630][__main__][INFO] - Epoch 9 with learning rate [0.0001] | |
| [2024-01-18 00:30:36,234][__main__][INFO] - loss : 0.21692666750649664 | |
| [2024-01-18 00:31:09,535][__main__][INFO] - Epoch 10 with learning rate [0.0001] | |
| [2024-01-18 00:56:54,932][__main__][INFO] - loss : 0.21269827534258345 | |
| [2024-01-18 00:57:26,649][__main__][INFO] - Epoch 11 with learning rate [0.0001] | |
| [2024-01-18 01:23:25,489][__main__][INFO] - loss : 0.20919359432657592 | |
| [2024-01-18 01:23:59,103][__main__][INFO] - Epoch 12 with learning rate [0.0001] | |
| [2024-01-18 01:49:46,540][__main__][INFO] - loss : 0.20483679519097045 | |
| [2024-01-18 01:50:19,357][__main__][INFO] - Epoch 13 with learning rate [0.0001] | |
| [2024-01-18 02:15:44,356][__main__][INFO] - loss : 0.20158936556180315 | |
| [2024-01-18 02:16:18,075][__main__][INFO] - Epoch 14 with learning rate [0.0001] | |
| [2024-01-18 02:41:55,475][__main__][INFO] - loss : 0.19783804734051214 | |
| [2024-01-18 02:42:27,379][__main__][INFO] - Epoch 15 with learning rate [0.0001] | |
| [2024-01-18 03:08:05,537][__main__][INFO] - loss : 0.19492025763293092 | |
| [2024-01-18 03:08:36,388][__main__][INFO] - Epoch 16 with learning rate [0.0001] | |
| [2024-01-18 03:34:16,130][__main__][INFO] - loss : 0.19140285502374169 | |
| [2024-01-18 03:34:52,056][__main__][INFO] - Epoch 17 with learning rate [0.0001] | |
| [2024-01-18 04:00:32,084][__main__][INFO] - loss : 0.18805657176673365 | |
| [2024-01-18 04:01:02,232][__main__][INFO] - Epoch 18 with learning rate [0.0001] | |
| [2024-01-18 04:26:36,151][__main__][INFO] - loss : 0.185264913658301 | |
| [2024-01-18 04:27:09,632][__main__][INFO] - Epoch 19 with learning rate [0.0001] | |
| [2024-01-18 04:52:43,178][__main__][INFO] - loss : 0.18217022760709117 | |
| [2024-01-18 04:53:15,053][__main__][INFO] - Epoch 20 with learning rate [0.0001] | |
| [2024-01-18 05:16:37,190][__main__][INFO] - loss : 0.17973323059330423 | |
| [2024-01-18 05:17:00,886][__main__][INFO] - Epoch 21 with learning rate [0.0001] | |
| [2024-01-18 05:38:29,839][__main__][INFO] - loss : 0.1770827673673628 | |
| [2024-01-18 05:38:53,403][__main__][INFO] - Epoch 22 with learning rate [0.0001] | |
| [2024-01-18 06:00:22,922][__main__][INFO] - loss : 0.17473657063643144 | |
| [2024-01-18 06:00:46,463][__main__][INFO] - Epoch 23 with learning rate [0.0001] | |
| [2024-01-18 06:22:15,708][__main__][INFO] - loss : 0.17264075693736475 | |
| [2024-01-18 06:22:38,862][__main__][INFO] - Epoch 24 with learning rate [0.0001] | |
| [2024-01-18 06:44:08,292][__main__][INFO] - loss : 0.17019510875393928 | |
| [2024-01-18 06:44:31,911][__main__][INFO] - Train time [hour] : 10.628406446443664 h | |
| [2024-01-18 06:44:31,912][__main__][INFO] - Results saved in /home/bezzam/LenslessPiCam/outputs/2024-01-17/20-06-45 | |