[2025-03-29 13:56:39,573][__main__][INFO] - Using GPU for training. Main device : cuda
[2025-03-29 13:56:43,851][__main__][INFO] - PSF shape : torch.Size([1, 135, 240, 3])
[2025-03-29 13:56:43,867][__main__][INFO] - PSF min : 0.0
[2025-03-29 13:56:43,873][__main__][INFO] - PSF max : 0.12337537109851837
[2025-03-29 13:56:43,873][__main__][INFO] - PSF dtype : torch.float32
[2025-03-29 13:56:43,889][__main__][INFO] - PSF norm : 1.7320507764816284
[2025-03-29 13:56:43,890][__main__][INFO] - Reconstruction a few images with ADMM...
[2025-03-29 13:56:44,498][__main__][INFO] - Cropped shape :  (135, 240, 3)
[2025-03-29 13:56:45,584][__main__][INFO] - Train test size : 24000
[2025-03-29 13:56:45,584][__main__][INFO] - Test test size : 999
[2025-03-29 13:56:45,673][__main__][INFO] - Training model with 8161269 parameters
[2025-03-29 13:56:45,674][__main__][INFO] - -- Post-process model with 8161248 parameters
[2025-03-29 13:56:45,674][__main__][INFO] - Setup time : 0.08948564529418945 s
[2025-03-29 13:56:45,674][__main__][INFO] - PSF shape : torch.Size([1, 135, 240, 3])
[2025-03-29 13:56:45,674][__main__][INFO] - Results saved in /root/LenslessPiCam/outputs/2025-03-29/13-56-23
[2025-03-29 13:57:35,093][__main__][INFO] - Epoch 0 with learning rate [0.0, 0.0]
[2025-03-29 14:30:16,228][__main__][INFO] - loss : 0.46259485757102486
[2025-03-29 14:31:03,359][__main__][INFO] - Epoch 1 with learning rate [8e-05, 8e-05]
[2025-03-29 15:04:11,177][__main__][INFO] - loss : 0.31885068424294405
[2025-03-29 15:04:58,426][__main__][INFO] - Epoch 2 with learning rate [9.975414512725057e-05, 9.975414512725057e-05]
[2025-03-29 15:38:08,317][__main__][INFO] - loss : 0.27944719273348695
[2025-03-29 15:38:55,771][__main__][INFO] - Epoch 3 with learning rate [9.86663298624003e-05, 9.86663298624003e-05]
[2025-03-29 16:12:00,483][__main__][INFO] - loss : 0.2606012739986183
[2025-03-29 16:12:47,796][__main__][INFO] - Epoch 4 with learning rate [9.672822322997305e-05, 9.672822322997305e-05]
[2025-03-29 16:45:57,095][__main__][INFO] - loss : 0.2487822362408034
[2025-03-29 16:46:44,417][__main__][INFO] - Epoch 5 with learning rate [9.397368756032445e-05, 9.397368756032445e-05]
[2025-03-29 17:19:51,963][__main__][INFO] - loss : 0.2412841075509787
[2025-03-29 17:20:38,954][__main__][INFO] - Epoch 6 with learning rate [9.045084971874738e-05, 9.045084971874738e-05]
[2025-03-29 17:53:46,161][__main__][INFO] - loss : 0.2337305447210873
[2025-03-29 17:54:33,481][__main__][INFO] - Epoch 7 with learning rate [8.622126023955446e-05, 8.622126023955446e-05]
[2025-03-29 18:27:46,059][__main__][INFO] - loss : 0.22772211192051578
[2025-03-29 18:28:33,006][__main__][INFO] - Epoch 8 with learning rate [8.135881792367686e-05, 8.135881792367686e-05]
[2025-03-29 19:01:37,033][__main__][INFO] - loss : 0.22224611699829522
[2025-03-29 19:02:23,480][__main__][INFO] - Epoch 9 with learning rate [7.594847868906076e-05, 7.594847868906076e-05]
[2025-03-29 19:35:29,324][__main__][INFO] - loss : 0.2172668660556278
[2025-03-29 19:36:16,193][__main__][INFO] - Epoch 10 with learning rate [7.008477123264848e-05, 7.008477123264848e-05]
[2025-03-29 20:09:22,252][__main__][INFO] - loss : 0.21187449455509516
[2025-03-29 20:10:09,265][__main__][INFO] - Epoch 11 with learning rate [6.387014543809223e-05, 6.387014543809223e-05]
[2025-03-29 20:43:10,732][__main__][INFO] - loss : 0.20817500577494513
[2025-03-29 20:43:57,991][__main__][INFO] - Epoch 12 with learning rate [5.74131823855921e-05, 5.74131823855921e-05]
[2025-03-29 21:17:03,127][__main__][INFO] - loss : 0.2043119238540532
[2025-03-29 21:17:50,625][__main__][INFO] - Epoch 13 with learning rate [5.0826697238317935e-05, 5.0826697238317935e-05]
[2025-03-29 21:50:54,707][__main__][INFO] - loss : 0.20016479102149587
[2025-03-29 21:51:41,987][__main__][INFO] - Epoch 14 with learning rate [4.4225768151520694e-05, 4.4225768151520694e-05]
[2025-03-29 22:24:44,352][__main__][INFO] - loss : 0.19616918724651192
[2025-03-29 22:25:31,484][__main__][INFO] - Epoch 15 with learning rate [3.772572564296005e-05, 3.772572564296005e-05]
[2025-03-29 22:58:32,374][__main__][INFO] - loss : 0.19216100977857836
[2025-03-29 22:59:19,611][__main__][INFO] - Epoch 16 with learning rate [3.144013755408895e-05, 3.144013755408895e-05]
[2025-03-29 23:32:20,173][__main__][INFO] - loss : 0.18882135972380643
[2025-03-29 23:33:07,656][__main__][INFO] - Epoch 17 with learning rate [2.547882480847461e-05, 2.547882480847461e-05]
[2025-03-30 00:06:12,536][__main__][INFO] - loss : 0.18523583800345683
[2025-03-30 00:06:59,742][__main__][INFO] - Epoch 18 with learning rate [1.9945942635848748e-05, 1.9945942635848748e-05]
[2025-03-30 00:40:03,674][__main__][INFO] - loss : 0.18233649141838146
[2025-03-30 00:40:50,938][__main__][INFO] - Epoch 19 with learning rate [1.4938160786375572e-05, 1.4938160786375572e-05]
[2025-03-30 01:13:54,015][__main__][INFO] - loss : 0.1796669936416048
[2025-03-30 01:14:41,090][__main__][INFO] - Epoch 20 with learning rate [1.0542974530180327e-05, 1.0542974530180327e-05]
[2025-03-30 01:47:49,603][__main__][INFO] - loss : 0.17756925339375842
[2025-03-30 01:48:36,757][__main__][INFO] - Epoch 21 with learning rate [6.837175952121306e-06, 6.837175952121306e-06]
[2025-03-30 02:21:39,145][__main__][INFO] - loss : 0.175861525410165
[2025-03-30 02:22:27,197][__main__][INFO] - Epoch 22 with learning rate [3.885512251130763e-06, 3.885512251130763e-06]
[2025-03-30 02:55:38,510][__main__][INFO] - loss : 0.1746500805305938
[2025-03-30 02:56:26,528][__main__][INFO] - Epoch 23 with learning rate [1.7395544861325718e-06, 1.7395544861325718e-06]
[2025-03-30 03:29:28,199][__main__][INFO] - loss : 0.173926534210642
[2025-03-30 03:30:15,371][__main__][INFO] - Epoch 24 with learning rate [4.367965336512403e-07, 4.367965336512403e-07]
[2025-03-30 04:03:21,687][__main__][INFO] - loss : 0.1735909561738375
[2025-03-30 04:04:09,360][__main__][INFO] - Train time [hour] : 14.122845769988166 h
[2025-03-30 04:04:09,361][__main__][INFO] - Results saved in /root/LenslessPiCam/outputs/2025-03-29/13-56-23