| [2024-01-17 07:26:47,044][__main__][INFO] - Using GPU for training. | |
| [2024-01-17 07:26:48,377][__main__][INFO] - Reconstruction a few images with ADMM... | |
| [2024-01-17 07:26:49,429][__main__][INFO] - Train test size : 24000 | |
| [2024-01-17 07:26:49,429][__main__][INFO] - Test test size : 999 | |
| [2024-01-17 07:26:49,539][__main__][INFO] - Training model with 17010705 parameters | |
| [2024-01-17 07:26:49,540][__main__][INFO] - Setup time : 0.11027836799621582 s | |
| [2024-01-17 07:26:49,540][__main__][INFO] - PSF shape : torch.Size([1, 135, 240, 3]) | |
| [2024-01-17 07:26:49,540][__main__][INFO] - Results saved in /home/bezzam/LenslessPiCam/outputs/2024-01-17/07-26-46 | |
| [2024-01-17 07:27:20,311][__main__][INFO] - Epoch 0 with learning rate [0.0001] | |
| [2024-01-17 07:50:32,797][__main__][INFO] - loss : 0.4456000145574413 | |
| [2024-01-17 07:51:01,463][__main__][INFO] - Epoch 1 with learning rate [0.0001] | |
| [2024-01-17 08:14:20,250][__main__][INFO] - loss : 0.3752540885309385 | |
| [2024-01-17 08:14:49,257][__main__][INFO] - Epoch 2 with learning rate [0.0001] | |
| [2024-01-17 08:38:07,870][__main__][INFO] - loss : 0.3614413690666361 | |
| [2024-01-17 08:38:36,438][__main__][INFO] - Epoch 3 with learning rate [0.0001] | |
| [2024-01-17 09:03:45,961][__main__][INFO] - loss : 0.3543593736886971 | |
| [2024-01-17 09:04:21,711][__main__][INFO] - Epoch 4 with learning rate [0.0001] | |
| [2024-01-17 09:30:30,934][__main__][INFO] - loss : 0.34833630931377385 | |
| [2024-01-17 09:31:06,059][__main__][INFO] - Epoch 5 with learning rate [0.0001] | |
| [2024-01-17 09:57:22,546][__main__][INFO] - loss : 0.3443675898412858 | |
| [2024-01-17 09:57:56,097][__main__][INFO] - Epoch 6 with learning rate [0.0001] | |
| [2024-01-17 10:22:11,770][__main__][INFO] - loss : 0.3419632758597527 | |
| [2024-01-17 10:22:42,332][__main__][INFO] - Epoch 7 with learning rate [0.0001] | |
| [2024-01-17 10:46:46,545][__main__][INFO] - loss : 0.3396190254837278 | |
| [2024-01-17 10:47:17,002][__main__][INFO] - Epoch 8 with learning rate [0.0001] | |
| [2024-01-17 11:11:27,544][__main__][INFO] - loss : 0.33685440245270676 | |
| [2024-01-17 11:11:58,102][__main__][INFO] - Epoch 9 with learning rate [0.0001] | |
| [2024-01-17 11:36:27,823][__main__][INFO] - loss : 0.3356172657757996 | |
| [2024-01-17 11:36:57,622][__main__][INFO] - Epoch 10 with learning rate [0.0001] | |
| [2024-01-17 12:00:56,490][__main__][INFO] - loss : 0.33292641883095114 | |
| [2024-01-17 12:01:22,894][__main__][INFO] - Epoch 11 with learning rate [0.0001] | |
| [2024-01-17 12:28:34,978][__main__][INFO] - loss : 0.33115439516305917 | |
| [2024-01-17 12:29:15,624][__main__][INFO] - Epoch 12 with learning rate [0.0001] | |
| [2024-01-17 12:56:52,746][__main__][INFO] - loss : 0.33041740947465026 | |
| [2024-01-17 12:57:29,627][__main__][INFO] - Epoch 13 with learning rate [0.0001] | |
| [2024-01-17 13:26:26,260][__main__][INFO] - loss : 0.3293761413594094 | |
| [2024-01-17 13:27:08,275][__main__][INFO] - Epoch 14 with learning rate [0.0001] | |
| [2024-01-17 13:56:38,111][__main__][INFO] - loss : 0.3278790179193024 | |
| [2024-01-17 13:57:19,313][__main__][INFO] - Epoch 15 with learning rate [0.0001] | |
| [2024-01-17 14:26:29,781][__main__][INFO] - loss : 0.32690351243813787 | |
| [2024-01-17 14:27:11,968][__main__][INFO] - Epoch 16 with learning rate [0.0001] | |
| [2024-01-17 14:56:33,234][__main__][INFO] - loss : 0.32551091246306796 | |
| [2024-01-17 14:57:15,742][__main__][INFO] - Epoch 17 with learning rate [0.0001] | |
| [2024-01-17 15:26:02,768][__main__][INFO] - loss : 0.3250028810103734 | |
| [2024-01-17 15:26:46,070][__main__][INFO] - Epoch 18 with learning rate [0.0001] | |
| [2024-01-17 15:55:57,763][__main__][INFO] - loss : 0.3241230170826122 | |
| [2024-01-17 15:56:39,317][__main__][INFO] - Epoch 19 with learning rate [0.0001] | |
| [2024-01-17 16:25:52,023][__main__][INFO] - loss : 0.3236212235540147 | |
| [2024-01-17 16:26:34,570][__main__][INFO] - Epoch 20 with learning rate [0.0001] | |
| [2024-01-17 16:56:12,012][__main__][INFO] - loss : 0.32267501302063534 | |
| [2024-01-17 16:57:00,362][__main__][INFO] - Epoch 21 with learning rate [0.0001] | |
| [2024-01-17 17:26:42,932][__main__][INFO] - loss : 0.32195841999352015 | |
| [2024-01-17 17:27:25,056][__main__][INFO] - Epoch 22 with learning rate [0.0001] | |
| [2024-01-17 17:57:34,972][__main__][INFO] - loss : 0.3214100161890192 | |
| [2024-01-17 17:58:15,599][__main__][INFO] - Epoch 23 with learning rate [0.0001] | |
| [2024-01-17 18:27:40,650][__main__][INFO] - loss : 0.32090955423812023 | |
| [2024-01-17 18:28:21,632][__main__][INFO] - Epoch 24 with learning rate [0.0001] | |
| [2024-01-17 18:57:20,342][__main__][INFO] - loss : 0.3196497395386296 | |
| [2024-01-17 18:58:02,462][__main__][INFO] - Train time [hour] : 11.520014075040818 h | |
| [2024-01-17 18:58:02,462][__main__][INFO] - Results saved in /home/bezzam/LenslessPiCam/outputs/2024-01-17/07-26-46 | |