| [2024-01-17 13:07:54,899][__main__][INFO] - Using GPU for training. | |
| [2024-01-17 13:07:56,670][__main__][INFO] - Reconstruction a few images with ADMM... | |
| [2024-01-17 13:07:58,131][__main__][INFO] - Train test size : 24000 | |
| [2024-01-17 13:07:58,131][__main__][INFO] - Test test size : 999 | |
| [2024-01-17 13:07:58,277][__main__][INFO] - Training model with 17010705 parameters | |
| [2024-01-17 13:07:58,277][__main__][INFO] - Setup time : 0.14583206176757812 s | |
| [2024-01-17 13:07:58,277][__main__][INFO] - PSF shape : torch.Size([1, 135, 240, 3]) | |
| [2024-01-17 13:07:58,278][__main__][INFO] - Results saved in /home/bezzam/LenslessPiCam/outputs/2024-01-17/13-07-54 | |
| [2024-01-17 13:09:13,355][__main__][INFO] - Epoch 0 with learning rate [0.0001] | |
| [2024-01-17 14:03:09,784][__main__][INFO] - loss : 0.476322088787954 | |
| [2024-01-17 14:04:16,494][__main__][INFO] - Epoch 1 with learning rate [0.0001] | |
| [2024-01-17 14:58:03,701][__main__][INFO] - loss : 0.40024586493770287 | |
| [2024-01-17 14:59:09,288][__main__][INFO] - Epoch 2 with learning rate [0.0001] | |
| [2024-01-17 15:52:23,719][__main__][INFO] - loss : 0.37944049969315513 | |
| [2024-01-17 15:53:35,887][__main__][INFO] - Epoch 3 with learning rate [0.0001] | |
| [2024-01-17 16:47:32,255][__main__][INFO] - loss : 0.36910764841238647 | |
| [2024-01-17 16:48:44,803][__main__][INFO] - Epoch 4 with learning rate [0.0001] | |
| [2024-01-17 17:43:22,669][__main__][INFO] - loss : 0.3623753864367814 | |
| [2024-01-17 17:44:36,551][__main__][INFO] - Epoch 5 with learning rate [0.0001] | |
| [2024-01-17 18:38:38,027][__main__][INFO] - loss : 0.3566511044204232 | |
| [2024-01-17 18:39:50,677][__main__][INFO] - Epoch 6 with learning rate [0.0001] | |
| [2024-01-17 19:28:51,966][__main__][INFO] - loss : 0.35289569496611717 | |
| [2024-01-17 19:29:41,815][__main__][INFO] - Epoch 7 with learning rate [0.0001] | |
| [2024-01-17 20:18:14,894][__main__][INFO] - loss : 0.3488064972112566 | |
| [2024-01-17 20:19:20,047][__main__][INFO] - Epoch 8 with learning rate [0.0001] | |
| [2024-01-17 21:12:51,459][__main__][INFO] - loss : 0.34573624481757476 | |
| [2024-01-17 21:13:58,487][__main__][INFO] - Epoch 9 with learning rate [0.0001] | |
| [2024-01-17 22:07:32,518][__main__][INFO] - loss : 0.3425459060768282 | |
| [2024-01-17 22:08:36,228][__main__][INFO] - Epoch 10 with learning rate [0.0001] | |
| [2024-01-17 23:01:50,048][__main__][INFO] - loss : 0.3399206795791783 | |
| [2024-01-17 23:02:54,902][__main__][INFO] - Epoch 11 with learning rate [0.0001] | |
| [2024-01-17 23:56:22,911][__main__][INFO] - loss : 0.336960077216228 | |
| [2024-01-17 23:57:26,207][__main__][INFO] - Epoch 12 with learning rate [0.0001] | |
| [2024-01-18 00:50:49,637][__main__][INFO] - loss : 0.3344712836692734 | |
| [2024-01-18 00:51:52,869][__main__][INFO] - Epoch 13 with learning rate [0.0001] | |
| [2024-01-18 01:45:13,939][__main__][INFO] - loss : 0.3317801366696754 | |
| [2024-01-18 01:46:19,391][__main__][INFO] - Epoch 14 with learning rate [0.0001] | |
| [2024-01-18 02:39:17,484][__main__][INFO] - loss : 0.3294699110239741 | |
| [2024-01-18 02:40:20,765][__main__][INFO] - Epoch 15 with learning rate [0.0001] | |
| [2024-01-18 03:33:33,902][__main__][INFO] - loss : 0.3273846674313151 | |
| [2024-01-18 03:34:41,336][__main__][INFO] - Epoch 16 with learning rate [0.0001] | |
| [2024-01-18 04:27:54,219][__main__][INFO] - loss : 0.3251873636494089 | |
| [2024-01-18 04:28:57,022][__main__][INFO] - Epoch 17 with learning rate [0.0001] | |
| [2024-01-18 05:12:09,120][__main__][INFO] - loss : 0.3231429106344786 | |
| [2024-01-18 05:12:37,868][__main__][INFO] - Epoch 18 with learning rate [0.0001] | |
| [2024-01-18 05:35:52,325][__main__][INFO] - loss : 0.32110762567818224 | |
| [2024-01-18 05:36:20,543][__main__][INFO] - Epoch 19 with learning rate [0.0001] | |
| [2024-01-18 05:59:34,016][__main__][INFO] - loss : 0.319129562412699 | |
| [2024-01-18 06:00:02,519][__main__][INFO] - Epoch 20 with learning rate [0.0001] | |
| [2024-01-18 06:23:17,839][__main__][INFO] - loss : 0.3169412605613476 | |
| [2024-01-18 06:23:46,171][__main__][INFO] - Epoch 21 with learning rate [0.0001] | |
| [2024-01-18 06:47:04,725][__main__][INFO] - loss : 0.31541262497007855 | |
| [2024-01-18 06:47:31,876][__main__][INFO] - Epoch 22 with learning rate [0.0001] | |
| [2024-01-18 07:11:25,750][__main__][INFO] - loss : 0.31347458137571854 | |
| [2024-01-18 07:11:53,205][__main__][INFO] - Epoch 23 with learning rate [0.0001] | |
| [2024-01-18 07:35:45,792][__main__][INFO] - loss : 0.3116545605609805 | |
| [2024-01-18 07:36:13,231][__main__][INFO] - Epoch 24 with learning rate [0.0001] | |
| [2024-01-18 08:00:09,494][__main__][INFO] - loss : 0.31003324522078024 | |
| [2024-01-18 08:00:36,420][__main__][INFO] - Train time [hour] : 18.876945508718492 h | |
| [2024-01-18 08:00:36,420][__main__][INFO] - Results saved in /home/bezzam/LenslessPiCam/outputs/2024-01-17/13-07-54 | |