| [2024-08-27 19:13:39,179][__main__][INFO] - Using GPU for training. Main device : cuda:0 | |
| [2024-08-27 19:13:48,613][__main__][INFO] - PSF shape : torch.Size([1, 380, 507, 3]) | |
| [2024-08-27 19:13:48,643][__main__][INFO] - PSF min : 0.0 | |
| [2024-08-27 19:13:48,655][__main__][INFO] - PSF max : 0.04469018802046776 | |
| [2024-08-27 19:13:48,655][__main__][INFO] - PSF dtype : torch.float32 | |
| [2024-08-27 19:13:48,697][__main__][INFO] - PSF norm : 1.000011682510376 | |
| [2024-08-27 19:13:48,698][__main__][INFO] - Reconstruction a few images with ADMM... | |
| [2024-08-27 19:13:49,464][__main__][INFO] - Cropped shape : (123, 123, 3) | |
| [2024-08-27 19:13:49,750][__main__][INFO] - Cropped shape : (123, 123, 3) | |
| [2024-08-27 19:13:52,245][__main__][INFO] - Train test size : 21250 | |
| [2024-08-27 19:13:52,246][__main__][INFO] - Test test size : 3750 | |
| [2024-08-27 19:13:52,357][__main__][INFO] - Training model with 8138610 parameters | |
| [2024-08-27 19:13:52,358][__main__][INFO] - -- Post-process model with 8010848 parameters | |
| [2024-08-27 19:13:52,358][__main__][INFO] - -- Background subtraction model with 127740 parameters | |
| [2024-08-27 19:13:52,358][__main__][INFO] - Setup time : 0.11179780960083008 s | |
| [2024-08-27 19:13:52,359][__main__][INFO] - PSF shape : torch.Size([1, 380, 507, 3]) | |
| [2024-08-27 19:13:52,359][__main__][INFO] - Results saved in /root/FORKS/LenslessPiCamStefan/outputs/2024-08-27/19-13-34 | |
| [2024-08-27 19:21:28,059][__main__][INFO] - Epoch 0 with learning rate [0.0, 0.0] | |
| [2024-08-27 20:56:47,128][__main__][INFO] - loss : 0.9927158449276344 | |
| [2024-08-27 21:04:22,288][__main__][INFO] - Epoch 1 with learning rate [0.0001, 0.0001] | |
| [2024-08-27 22:39:25,430][__main__][INFO] - loss : 0.5123431228680422 | |
| [2024-08-27 22:47:01,462][__main__][INFO] - Epoch 2 with learning rate [9.957224306869053e-05, 9.957224306869053e-05] | |
| [2024-08-28 00:21:59,187][__main__][INFO] - loss : 0.4371124910653122 | |
| [2024-08-28 00:29:35,474][__main__][INFO] - Epoch 3 with learning rate [9.829629131445342e-05, 9.829629131445342e-05] | |
| [2024-08-28 02:04:29,951][__main__][INFO] - loss : 0.4098805369135799 | |
| [2024-08-28 02:12:06,029][__main__][INFO] - Epoch 4 with learning rate [9.619397662556435e-05, 9.619397662556435e-05] | |
| [2024-08-28 03:47:02,919][__main__][INFO] - loss : 0.39318362969408044 | |
| [2024-08-28 03:54:38,101][__main__][INFO] - Epoch 5 with learning rate [9.330127018922194e-05, 9.330127018922194e-05] | |
| [2024-08-28 05:29:32,131][__main__][INFO] - loss : 0.3808138402543146 | |
| [2024-08-28 05:37:08,469][__main__][INFO] - Epoch 6 with learning rate [8.966766701456177e-05, 8.966766701456177e-05] | |
| [2024-08-28 07:12:01,763][__main__][INFO] - loss : 0.371333374325086 | |
| [2024-08-28 07:19:37,242][__main__][INFO] - Epoch 7 with learning rate [8.535533905932738e-05, 8.535533905932738e-05] | |
| [2024-08-28 08:54:33,723][__main__][INFO] - loss : 0.36291745473901943 | |
| [2024-08-28 09:02:08,737][__main__][INFO] - Epoch 8 with learning rate [8.043807145043604e-05, 8.043807145043604e-05] | |
| [2024-08-28 10:37:10,317][__main__][INFO] - loss : 0.355292763690727 | |
| [2024-08-28 10:44:50,889][__main__][INFO] - Epoch 9 with learning rate [7.500000000000001e-05, 7.500000000000001e-05] | |
| [2024-08-28 12:19:48,252][__main__][INFO] - loss : 0.3481771532714702 | |
| [2024-08-28 12:27:22,419][__main__][INFO] - Epoch 10 with learning rate [6.91341716182545e-05, 6.91341716182545e-05] | |
| [2024-08-28 14:02:16,114][__main__][INFO] - loss : 0.34093639172673873 | |
| [2024-08-28 14:09:50,227][__main__][INFO] - Epoch 11 with learning rate [6.294095225512603e-05, 6.294095225512603e-05] | |
| [2024-08-28 15:43:48,593][__main__][INFO] - loss : 0.3343457742843967 | |
| [2024-08-28 15:51:19,946][__main__][INFO] - Epoch 12 with learning rate [5.6526309611002594e-05, 5.6526309611002594e-05] | |
| [2024-08-28 17:25:09,081][__main__][INFO] - loss : 0.32738822705043935 | |
| [2024-08-28 17:32:39,176][__main__][INFO] - Epoch 13 with learning rate [5e-05, 5e-05] | |
| [2024-08-28 19:06:34,306][__main__][INFO] - loss : 0.32103165649277343 | |
| [2024-08-28 19:14:05,597][__main__][INFO] - Epoch 14 with learning rate [4.347369038899744e-05, 4.347369038899744e-05] | |
| [2024-08-28 20:47:57,809][__main__][INFO] - loss : 0.3145939674548859 | |
| [2024-08-28 20:55:29,207][__main__][INFO] - Epoch 15 with learning rate [3.705904774487396e-05, 3.705904774487396e-05] | |
| [2024-08-28 22:29:20,311][__main__][INFO] - loss : 0.3083602398712487 | |
| [2024-08-28 22:36:52,719][__main__][INFO] - Epoch 16 with learning rate [3.086582838174551e-05, 3.086582838174551e-05] | |
| [2024-08-29 00:10:48,033][__main__][INFO] - loss : 0.30239493312212784 | |
| [2024-08-29 00:18:20,232][__main__][INFO] - Epoch 17 with learning rate [2.500000000000001e-05, 2.500000000000001e-05] | |
| [2024-08-29 01:52:18,799][__main__][INFO] - loss : 0.2970171544498207 | |
| [2024-08-29 01:59:53,169][__main__][INFO] - Epoch 18 with learning rate [1.9561928549563968e-05, 1.9561928549563968e-05] | |
| [2024-08-29 03:33:52,122][__main__][INFO] - loss : 0.292060820257149 | |
| [2024-08-29 03:41:24,143][__main__][INFO] - Epoch 19 with learning rate [1.4644660940672627e-05, 1.4644660940672627e-05] | |
| [2024-08-29 05:15:20,185][__main__][INFO] - loss : 0.2877190843373178 | |
| [2024-08-29 05:22:52,413][__main__][INFO] - Epoch 20 with learning rate [1.0332332985438248e-05, 1.0332332985438248e-05] | |
| [2024-08-29 06:56:47,094][__main__][INFO] - loss : 0.2840360306019672 | |
| [2024-08-29 07:04:20,785][__main__][INFO] - Epoch 21 with learning rate [6.698729810778065e-06, 6.698729810778065e-06] | |
| [2024-08-29 08:38:18,524][__main__][INFO] - loss : 0.2811169280897917 | |
| [2024-08-29 08:45:51,509][__main__][INFO] - Epoch 22 with learning rate [3.8060233744356633e-06, 3.8060233744356633e-06] | |
| [2024-08-29 10:19:52,840][__main__][INFO] - loss : 0.27893567653770796 | |
| [2024-08-29 10:27:26,479][__main__][INFO] - Epoch 23 with learning rate [1.70370868554659e-06, 1.70370868554659e-06] | |
| [2024-08-29 12:01:21,790][__main__][INFO] - loss : 0.2774795897987941 | |
| [2024-08-29 12:08:55,322][__main__][INFO] - Epoch 24 with learning rate [4.277569313094809e-07, 4.277569313094809e-07] | |
| [2024-08-29 13:42:55,017][__main__][INFO] - loss : 0.2766513896514619 | |
| [2024-08-29 13:50:28,107][__main__][INFO] - Train time [hour] : 42.60953488018777 h | |
| [2024-08-29 13:50:28,107][__main__][INFO] - Results saved in /root/FORKS/LenslessPiCamStefan/outputs/2024-08-27/19-13-34 | |