train_learning_based.log

[2024-08-24 12:35:41,572][__main__][INFO] - Using GPU for training. Main device : cuda:0
[2024-08-24 12:37:42,715][__main__][INFO] - PSF shape : torch.Size([1, 380, 507, 3])
[2024-08-24 12:37:42,733][__main__][INFO] - PSF min : 0.0
[2024-08-24 12:37:42,737][__main__][INFO] - PSF max : 0.04469018802046776
[2024-08-24 12:37:42,738][__main__][INFO] - PSF dtype : torch.float32
[2024-08-24 12:37:42,761][__main__][INFO] - PSF norm : 1.000011682510376
[2024-08-24 12:37:42,761][__main__][INFO] - Reconstruction a few images with ADMM...
[2024-08-24 12:37:43,366][__main__][INFO] - Cropped shape :  (123, 123, 3)
[2024-08-24 12:37:43,591][__main__][INFO] - Cropped shape :  (123, 123, 3)
[2024-08-24 12:37:45,721][__main__][INFO] - Train test size : 21250
[2024-08-24 12:37:45,721][__main__][INFO] - Test test size : 3750
[2024-08-24 12:37:45,822][__main__][INFO] - Training model with 8123453 parameters
[2024-08-24 12:37:45,823][__main__][INFO] - -- Post-process model with 8123432 parameters
[2024-08-24 12:37:45,823][__main__][INFO] - Setup time : 0.10174393653869629 s
[2024-08-24 12:37:45,823][__main__][INFO] - PSF shape : torch.Size([1, 380, 507, 3])
[2024-08-24 12:37:45,823][__main__][INFO] - Results saved in /root/FORKS/LenslessPiCamStefan/outputs/2024-08-24/12-35-36
[2024-08-24 12:45:03,746][__main__][INFO] - Epoch 0 with learning rate [0.0, 0.0]
[2024-08-24 14:11:57,590][__main__][INFO] - loss : 1.0008448002232089
[2024-08-24 14:19:13,903][__main__][INFO] - Epoch 1 with learning rate [0.0001, 0.0001]
[2024-08-24 15:45:50,134][__main__][INFO] - loss : 0.6258369574669302
[2024-08-24 15:53:07,356][__main__][INFO] - Epoch 2 with learning rate [9.957224306869053e-05, 9.957224306869053e-05]
[2024-08-24 17:19:46,730][__main__][INFO] - loss : 0.5080787176615114
[2024-08-24 17:27:02,544][__main__][INFO] - Epoch 3 with learning rate [9.829629131445342e-05, 9.829629131445342e-05]
[2024-08-24 18:53:36,162][__main__][INFO] - loss : 0.4745524097595114
[2024-08-24 19:00:52,229][__main__][INFO] - Epoch 4 with learning rate [9.619397662556435e-05, 9.619397662556435e-05]
[2024-08-24 20:27:31,921][__main__][INFO] - loss : 0.45491890771654175
[2024-08-24 20:34:53,678][__main__][INFO] - Epoch 5 with learning rate [9.330127018922194e-05, 9.330127018922194e-05]
[2024-08-24 22:01:32,096][__main__][INFO] - loss : 0.43754031682068345
[2024-08-24 22:08:48,143][__main__][INFO] - Epoch 6 with learning rate [8.966766701456177e-05, 8.966766701456177e-05]
[2024-08-24 23:35:26,348][__main__][INFO] - loss : 0.4248277737655456
[2024-08-24 23:42:40,958][__main__][INFO] - Epoch 7 with learning rate [8.535533905932738e-05, 8.535533905932738e-05]
[2024-08-25 01:09:14,486][__main__][INFO] - loss : 0.41310470932265525
[2024-08-25 01:16:30,892][__main__][INFO] - Epoch 8 with learning rate [8.043807145043604e-05, 8.043807145043604e-05]
[2024-08-25 02:43:03,239][__main__][INFO] - loss : 0.4052799641609231
[2024-08-25 02:50:19,985][__main__][INFO] - Epoch 9 with learning rate [7.500000000000001e-05, 7.500000000000001e-05]
[2024-08-25 04:16:55,556][__main__][INFO] - loss : 0.3955657965726784
[2024-08-25 04:24:15,568][__main__][INFO] - Epoch 10 with learning rate [6.91341716182545e-05, 6.91341716182545e-05]
[2024-08-25 05:50:55,154][__main__][INFO] - loss : 0.3874396745619648
[2024-08-25 05:58:11,400][__main__][INFO] - Epoch 11 with learning rate [6.294095225512603e-05, 6.294095225512603e-05]
[2024-08-25 07:24:49,705][__main__][INFO] - loss : 0.3800189991062844
[2024-08-25 07:32:05,599][__main__][INFO] - Epoch 12 with learning rate [5.6526309611002594e-05, 5.6526309611002594e-05]
[2024-08-25 08:58:43,151][__main__][INFO] - loss : 0.37254684900420737
[2024-08-25 09:06:04,700][__main__][INFO] - Epoch 13 with learning rate [5e-05, 5e-05]
[2024-08-25 10:32:44,832][__main__][INFO] - loss : 0.3661267356461512
[2024-08-25 10:40:01,603][__main__][INFO] - Epoch 14 with learning rate [4.347369038899744e-05, 4.347369038899744e-05]
[2024-08-25 12:06:41,963][__main__][INFO] - loss : 0.358960236928433
[2024-08-25 12:14:03,265][__main__][INFO] - Epoch 15 with learning rate [3.705904774487396e-05, 3.705904774487396e-05]
[2024-08-25 13:40:41,759][__main__][INFO] - loss : 0.35281407743350157
[2024-08-25 13:47:58,718][__main__][INFO] - Epoch 16 with learning rate [3.086582838174551e-05, 3.086582838174551e-05]
[2024-08-25 15:14:37,479][__main__][INFO] - loss : 0.34612425465368923
[2024-08-25 15:21:53,844][__main__][INFO] - Epoch 17 with learning rate [2.500000000000001e-05, 2.500000000000001e-05]
[2024-08-25 16:48:30,755][__main__][INFO] - loss : 0.3404686063522444
[2024-08-25 16:55:46,607][__main__][INFO] - Epoch 18 with learning rate [1.9561928549563968e-05, 1.9561928549563968e-05]
[2024-08-25 18:22:24,634][__main__][INFO] - loss : 0.3356301861194805
[2024-08-25 18:29:40,940][__main__][INFO] - Epoch 19 with learning rate [1.4644660940672627e-05, 1.4644660940672627e-05]
[2024-08-25 19:56:19,959][__main__][INFO] - loss : 0.3307931246493643
[2024-08-25 20:03:39,347][__main__][INFO] - Epoch 20 with learning rate [1.0332332985438248e-05, 1.0332332985438248e-05]
[2024-08-25 21:30:23,795][__main__][INFO] - loss : 0.3267237345686113
[2024-08-25 21:37:40,677][__main__][INFO] - Epoch 21 with learning rate [6.698729810778065e-06, 6.698729810778065e-06]
[2024-08-25 23:04:19,872][__main__][INFO] - loss : 0.3235375798233211
[2024-08-25 23:11:35,868][__main__][INFO] - Epoch 22 with learning rate [3.8060233744356633e-06, 3.8060233744356633e-06]
[2024-08-26 00:38:13,743][__main__][INFO] - loss : 0.3212229136707823
[2024-08-26 00:45:31,429][__main__][INFO] - Epoch 23 with learning rate [1.70370868554659e-06, 1.70370868554659e-06]
[2024-08-26 02:12:14,034][__main__][INFO] - loss : 0.3196195880841261
[2024-08-26 02:19:30,429][__main__][INFO] - Epoch 24 with learning rate [4.277569313094809e-07, 4.277569313094809e-07]
[2024-08-26 03:46:09,373][__main__][INFO] - loss : 0.3187509000065778
[2024-08-26 03:53:29,118][__main__][INFO] - Train time [hour] : 39.26164247320758 h
[2024-08-26 03:53:29,118][__main__][INFO] - Results saved in /root/FORKS/LenslessPiCamStefan/outputs/2024-08-24/12-35-36