[2024-03-19 23:05:13,899][__main__][INFO] - Using GPU for training.
[2024-03-19 23:05:27,156][__main__][INFO] - PSF shape : torch.Size([1, 380, 507, 3])
[2024-03-19 23:05:27,175][__main__][INFO] - PSF min : 0.0
[2024-03-19 23:05:27,189][__main__][INFO] - PSF max : 0.00769695034250617
[2024-03-19 23:05:27,189][__main__][INFO] - PSF dtype : torch.float32
[2024-03-19 23:05:27,226][__main__][INFO] - PSF norm : 1.0000050067901611
[2024-03-19 23:05:27,227][__main__][INFO] - Reconstruction a few images with ADMM...
[2024-03-19 23:05:28,177][__main__][INFO] - Cropped shape :  (262, 215, 3)
[2024-03-19 23:05:30,188][__main__][INFO] - Train test size : 22100
[2024-03-19 23:05:30,189][__main__][INFO] - Test test size : 3900
[2024-03-19 23:05:30,311][__main__][INFO] - Training model with 8739229 parameters
[2024-03-19 23:05:30,312][__main__][INFO] - Setup time : 0.12295293807983398 s
[2024-03-19 23:05:30,312][__main__][INFO] - PSF shape : torch.Size([1, 380, 507, 3])
[2024-03-19 23:05:30,312][__main__][INFO] - Results saved in /root/LenslessPiCam/outputs/2024-03-19/23-05-13
[2024-03-19 23:11:29,566][__main__][INFO] - Epoch 0 with learning rate [0.0001]
[2024-03-20 00:29:34,171][__main__][INFO] - loss : 0.6027056292171394
[2024-03-20 00:37:24,415][__main__][INFO] - Epoch 1 with learning rate [0.0001]
[2024-03-20 02:14:57,834][__main__][INFO] - loss : 0.45491452376227587
[2024-03-20 02:23:00,828][__main__][INFO] - Epoch 2 with learning rate [0.0001]
[2024-03-20 03:59:59,170][__main__][INFO] - loss : 0.4109719148048991
[2024-03-20 04:08:24,537][__main__][INFO] - Epoch 3 with learning rate [0.0001]
[2024-03-20 05:45:16,979][__main__][INFO] - loss : 0.39607362787108635
[2024-03-20 05:53:44,640][__main__][INFO] - Epoch 4 with learning rate [0.0001]
[2024-03-20 07:30:35,201][__main__][INFO] - loss : 0.38655266137144667
[2024-03-20 07:39:02,131][__main__][INFO] - Epoch 5 with learning rate [0.0001]
[2024-03-20 09:15:58,993][__main__][INFO] - loss : 0.37914496809648734
[2024-03-20 09:24:26,821][__main__][INFO] - Epoch 6 with learning rate [0.0001]
[2024-03-20 11:01:33,561][__main__][INFO] - loss : 0.37272920691319855
[2024-03-20 11:10:00,128][__main__][INFO] - Epoch 7 with learning rate [0.0001]
[2024-03-20 12:46:57,374][__main__][INFO] - loss : 0.3685111295961149
[2024-03-20 12:55:23,317][__main__][INFO] - Epoch 8 with learning rate [0.0001]
[2024-03-20 14:32:10,899][__main__][INFO] - loss : 0.3647502618704448
[2024-03-20 14:40:38,042][__main__][INFO] - Epoch 9 with learning rate [0.0001]
[2024-03-20 16:17:31,795][__main__][INFO] - loss : 0.3602015723497058
[2024-03-20 16:25:57,746][__main__][INFO] - Epoch 10 with learning rate [0.0001]
[2024-03-20 18:02:52,220][__main__][INFO] - loss : 0.35629392379819036
[2024-03-20 18:11:16,174][__main__][INFO] - Epoch 11 with learning rate [0.0001]
[2024-03-20 19:48:12,260][__main__][INFO] - loss : 0.35387199916990575
[2024-03-20 19:56:37,367][__main__][INFO] - Epoch 12 with learning rate [0.0001]
[2024-03-20 21:20:12,578][__main__][INFO] - loss : 0.35092777077968246
[2024-03-20 21:26:12,737][__main__][INFO] - Epoch 13 with learning rate [0.0001]
[2024-03-20 22:25:15,700][__main__][INFO] - loss : 0.3481461893783977
[2024-03-20 22:31:52,904][__main__][INFO] - Epoch 14 with learning rate [0.0001]
[2024-03-20 23:33:38,460][__main__][INFO] - loss : 0.34550664402241055
[2024-03-20 23:39:47,257][__main__][INFO] - Epoch 15 with learning rate [0.0001]
[2024-03-21 00:59:17,288][__main__][INFO] - loss : 0.3431852617576648
[2024-03-21 01:06:58,155][__main__][INFO] - Epoch 16 with learning rate [0.0001]
[2024-03-21 02:26:44,817][__main__][INFO] - loss : 0.34100570388778745
[2024-03-21 02:34:27,009][__main__][INFO] - Epoch 17 with learning rate [0.0001]
[2024-03-21 03:54:29,127][__main__][INFO] - loss : 0.3389040033337219
[2024-03-21 04:02:13,152][__main__][INFO] - Epoch 18 with learning rate [0.0001]
[2024-03-21 05:21:57,412][__main__][INFO] - loss : 0.33692646985410035
[2024-03-21 05:29:34,184][__main__][INFO] - Epoch 19 with learning rate [0.0001]
[2024-03-21 06:48:51,311][__main__][INFO] - loss : 0.33530701721146167
[2024-03-21 06:56:03,133][__main__][INFO] - Epoch 20 with learning rate [0.0001]
[2024-03-21 08:09:34,981][__main__][INFO] - loss : 0.33296261244499925
[2024-03-21 08:16:15,380][__main__][INFO] - Epoch 21 with learning rate [0.0001]
[2024-03-21 09:29:53,059][__main__][INFO] - loss : 0.33138960044038934
[2024-03-21 09:37:06,149][__main__][INFO] - Epoch 22 with learning rate [0.0001]
[2024-03-21 10:50:39,642][__main__][INFO] - loss : 0.3294696885320413
[2024-03-21 10:57:50,557][__main__][INFO] - Epoch 23 with learning rate [0.0001]
[2024-03-21 12:11:18,670][__main__][INFO] - loss : 0.3280403632087413
[2024-03-21 12:18:34,786][__main__][INFO] - Epoch 24 with learning rate [0.0001]
[2024-03-21 13:32:16,673][__main__][INFO] - loss : 0.32668002327373175
[2024-03-21 13:39:29,008][__main__][INFO] - Train time [hour] : 38.56589601073 h
[2024-03-21 13:39:29,008][__main__][INFO] - Results saved in /root/LenslessPiCam/outputs/2024-03-19/23-05-13