| [2024-11-18 21:44:00,651][__main__][INFO] - Using GPU for training. Main device : cuda:1 | |
| [2024-11-18 21:44:05,496][__main__][INFO] - PSF shape : torch.Size([1, 135, 240, 3]) | |
| [2024-11-18 21:44:05,511][__main__][INFO] - PSF min : -0.20868328213691711 | |
| [2024-11-18 21:44:05,522][__main__][INFO] - PSF max : 0.22161588072776794 | |
| [2024-11-18 21:44:05,522][__main__][INFO] - PSF dtype : torch.float32 | |
| [2024-11-18 21:44:05,557][__main__][INFO] - PSF norm : 15.264852523803711 | |
| [2024-11-18 21:44:05,557][__main__][INFO] - Reconstruction a few images with ADMM... | |
| [2024-11-18 21:44:06,237][__main__][INFO] - Cropped shape : (135, 240, 3) | |
| [2024-11-18 21:44:07,657][__main__][INFO] - Train test size : 24000 | |
| [2024-11-18 21:44:07,658][__main__][INFO] - Test test size : 999 | |
| [2024-11-18 21:44:07,775][__main__][INFO] - Training model with 8102358 parameters | |
| [2024-11-18 21:44:07,775][__main__][INFO] - -- Pre-process model with 4051168 parameters | |
| [2024-11-18 21:44:07,776][__main__][INFO] - -- Post-process model with 4051168 parameters | |
| [2024-11-18 21:44:07,776][__main__][INFO] - Setup time : 0.1182546615600586 s | |
| [2024-11-18 21:44:07,776][__main__][INFO] - PSF shape : torch.Size([1, 135, 240, 3]) | |
| [2024-11-18 21:44:07,776][__main__][INFO] - Results saved in /root/LenslessPiCam/outputs/2024-11-18/21-43-54 | |
| [2024-11-18 21:45:08,934][__main__][INFO] - Epoch 0 with learning rate [0.0001] | |
| [2024-11-18 22:31:00,400][__main__][INFO] - loss : 0.4457320999999838 | |
| [2024-11-18 22:32:02,187][__main__][INFO] - Epoch 1 with learning rate [0.0001] | |
| [2024-11-18 23:18:16,936][__main__][INFO] - loss : 0.35304422244677974 | |
| [2024-11-18 23:19:18,484][__main__][INFO] - Epoch 2 with learning rate [0.0001] | |
| [2024-11-19 00:05:51,894][__main__][INFO] - loss : 0.3186257427309945 | |
| [2024-11-19 00:06:52,709][__main__][INFO] - Epoch 3 with learning rate [0.0001] | |
| [2024-11-19 00:52:57,027][__main__][INFO] - loss : 0.3001118877554939 | |
| [2024-11-19 00:53:58,684][__main__][INFO] - Epoch 4 with learning rate [0.0001] | |
| [2024-11-19 01:40:21,355][__main__][INFO] - loss : 0.2871197715500995 | |
| [2024-11-19 01:41:22,854][__main__][INFO] - Epoch 5 with learning rate [0.0001] | |
| [2024-11-19 02:27:35,143][__main__][INFO] - loss : 0.2759193549056861 | |
| [2024-11-19 02:28:37,832][__main__][INFO] - Epoch 6 with learning rate [0.0001] | |
| [2024-11-19 03:14:54,771][__main__][INFO] - loss : 0.2672060283025112 | |
| [2024-11-19 03:15:56,602][__main__][INFO] - Epoch 7 with learning rate [0.0001] | |
| [2024-11-19 04:02:14,602][__main__][INFO] - loss : 0.26013692331065763 | |
| [2024-11-19 04:03:16,331][__main__][INFO] - Epoch 8 with learning rate [0.0001] | |
| [2024-11-19 04:49:28,989][__main__][INFO] - loss : 0.2529433202842869 | |
| [2024-11-19 04:50:31,044][__main__][INFO] - Epoch 9 with learning rate [0.0001] | |
| [2024-11-19 05:35:14,821][__main__][INFO] - loss : 0.24637296068668388 | |
| [2024-11-19 05:36:11,290][__main__][INFO] - Epoch 10 with learning rate [0.0001] | |
| [2024-11-19 06:12:50,192][__main__][INFO] - loss : 0.241643066485723 | |
| [2024-11-19 06:13:44,715][__main__][INFO] - Epoch 11 with learning rate [0.0001] | |
| [2024-11-19 06:49:51,964][__main__][INFO] - loss : 0.23715014848361418 | |
| [2024-11-19 06:50:45,998][__main__][INFO] - Epoch 12 with learning rate [0.0001] | |
| [2024-11-19 07:26:54,170][__main__][INFO] - loss : 0.23309148204326627 | |
| [2024-11-19 07:27:48,544][__main__][INFO] - Epoch 13 with learning rate [0.0001] | |
| [2024-11-19 08:03:54,069][__main__][INFO] - loss : 0.22964431804666907 | |
| [2024-11-19 08:04:48,307][__main__][INFO] - Epoch 14 with learning rate [0.0001] | |
| [2024-11-19 08:40:54,013][__main__][INFO] - loss : 0.22582625494400613 | |
| [2024-11-19 08:41:48,781][__main__][INFO] - Epoch 15 with learning rate [0.0001] | |
| [2024-11-19 09:17:54,471][__main__][INFO] - loss : 0.2224425456648071 | |
| [2024-11-19 09:18:47,655][__main__][INFO] - Epoch 16 with learning rate [0.0001] | |
| [2024-11-19 09:54:50,602][__main__][INFO] - loss : 0.21994397298619187 | |
| [2024-11-19 09:55:44,769][__main__][INFO] - Epoch 17 with learning rate [0.0001] | |
| [2024-11-19 10:31:46,876][__main__][INFO] - loss : 0.21696896877263985 | |
| [2024-11-19 10:32:40,340][__main__][INFO] - Epoch 18 with learning rate [0.0001] | |
| [2024-11-19 11:08:42,365][__main__][INFO] - loss : 0.21416821533441543 | |
| [2024-11-19 11:09:37,009][__main__][INFO] - Epoch 19 with learning rate [0.0001] | |
| [2024-11-19 11:45:46,488][__main__][INFO] - loss : 0.21184813575694983 | |
| [2024-11-19 11:46:40,907][__main__][INFO] - Epoch 20 with learning rate [0.0001] | |
| [2024-11-19 12:22:49,397][__main__][INFO] - loss : 0.2092782211887341 | |
| [2024-11-19 12:23:43,127][__main__][INFO] - Epoch 21 with learning rate [0.0001] | |
| [2024-11-19 12:59:53,371][__main__][INFO] - loss : 0.20700861246635466 | |
| [2024-11-19 13:00:47,243][__main__][INFO] - Epoch 22 with learning rate [0.0001] | |
| [2024-11-19 13:36:55,833][__main__][INFO] - loss : 0.2047655131593348 | |
| [2024-11-19 13:37:49,765][__main__][INFO] - Epoch 23 with learning rate [0.0001] | |
| [2024-11-19 14:13:59,622][__main__][INFO] - loss : 0.20259534879898072 | |
| [2024-11-19 14:14:53,609][__main__][INFO] - Epoch 24 with learning rate [0.0001] | |
| [2024-11-19 14:51:01,135][__main__][INFO] - loss : 0.2008559078400326 | |
| [2024-11-19 14:51:55,379][__main__][INFO] - Train time [hour] : 17.12942146572802 h | |
| [2024-11-19 14:51:55,380][__main__][INFO] - Results saved in /root/LenslessPiCam/outputs/2024-11-18/21-43-54 | |