[2024-03-24 15:35:50,128][__main__][INFO] - Using GPU for training.
[2024-03-24 15:36:03,708][__main__][INFO] - PSF shape : torch.Size([1, 380, 507, 3])
[2024-03-24 15:36:03,729][__main__][INFO] - PSF min : 0.0
[2024-03-24 15:36:03,741][__main__][INFO] - PSF max : 0.00769695034250617
[2024-03-24 15:36:03,742][__main__][INFO] - PSF dtype : torch.float32
[2024-03-24 15:36:03,779][__main__][INFO] - PSF norm : 1.0000050067901611
[2024-03-24 15:36:03,779][__main__][INFO] - Reconstruction a few images with ADMM...
[2024-03-24 15:36:04,721][__main__][INFO] - Cropped shape :  (262, 215, 3)
[2024-03-24 15:36:06,734][__main__][INFO] - Train test size : 22100
[2024-03-24 15:36:06,734][__main__][INFO] - Test test size : 3900
[2024-03-24 15:36:06,837][__main__][INFO] - Training model with 8102378 parameters
[2024-03-24 15:36:06,837][__main__][INFO] - Setup time : 0.10327005386352539 s
[2024-03-24 15:36:06,837][__main__][INFO] - PSF shape : torch.Size([1, 380, 507, 3])
[2024-03-24 15:36:06,838][__main__][INFO] - Results saved in /root/LenslessPiCam/outputs/2024-03-24/15-35-50
[2024-03-24 15:46:13,433][__main__][INFO] - Epoch 0 with learning rate [0.0001]
[2024-03-24 17:53:53,949][__main__][INFO] - loss : 0.45940045606496593
[2024-03-24 18:03:47,973][__main__][INFO] - Epoch 1 with learning rate [0.0001]
[2024-03-24 20:12:35,015][__main__][INFO] - loss : 0.35418758835997494
[2024-03-24 20:23:42,473][__main__][INFO] - Epoch 2 with learning rate [0.0001]
[2024-03-24 22:35:25,104][__main__][INFO] - loss : 0.3344209021599574
[2024-03-24 22:46:16,110][__main__][INFO] - Epoch 3 with learning rate [0.0001]
[2024-03-25 01:09:51,454][__main__][INFO] - loss : 0.3212876485106084
[2024-03-25 01:20:27,326][__main__][INFO] - Epoch 4 with learning rate [0.0001]
[2024-03-25 03:43:43,223][__main__][INFO] - loss : 0.31128515298549964
[2024-03-25 03:54:27,605][__main__][INFO] - Epoch 5 with learning rate [0.0001]
[2024-03-25 06:13:37,328][__main__][INFO] - loss : 0.303383419672827
[2024-03-25 06:23:30,744][__main__][INFO] - Epoch 6 with learning rate [0.0001]
[2024-03-25 08:32:59,741][__main__][INFO] - loss : 0.2966259348770072
[2024-03-25 08:43:53,790][__main__][INFO] - Epoch 7 with learning rate [0.0001]
[2024-03-25 11:06:56,738][__main__][INFO] - loss : 0.2915945121173941
[2024-03-25 11:17:36,381][__main__][INFO] - Epoch 8 with learning rate [0.0001]
[2024-03-25 13:32:04,408][__main__][INFO] - loss : 0.28663034119757075
[2024-03-25 13:42:24,201][__main__][INFO] - Epoch 9 with learning rate [0.0001]
[2024-03-25 15:42:51,017][__main__][INFO] - loss : 0.2827853759354585
[2024-03-25 15:52:41,137][__main__][INFO] - Epoch 10 with learning rate [0.0001]
[2024-03-25 17:58:16,188][__main__][INFO] - loss : 0.2793643888790686
[2024-03-25 18:08:23,417][__main__][INFO] - Epoch 11 with learning rate [0.0001]
[2024-03-25 20:19:13,547][__main__][INFO] - loss : 0.2763711688556285
[2024-03-25 20:28:49,588][__main__][INFO] - Epoch 12 with learning rate [0.0001]
[2024-03-25 22:31:44,351][__main__][INFO] - loss : 0.2737446634936664
[2024-03-25 22:41:33,394][__main__][INFO] - Epoch 13 with learning rate [0.0001]
[2024-03-26 00:43:46,076][__main__][INFO] - loss : 0.2711911531783866
[2024-03-26 00:53:53,085][__main__][INFO] - Epoch 14 with learning rate [0.0001]
[2024-03-26 03:03:18,809][__main__][INFO] - loss : 0.26922018789327934
[2024-03-26 03:14:06,642][__main__][INFO] - Epoch 15 with learning rate [0.0001]
[2024-03-26 05:24:39,620][__main__][INFO] - loss : 0.26704318030926977
[2024-03-26 05:35:22,490][__main__][INFO] - Epoch 16 with learning rate [0.0001]
[2024-03-26 07:46:05,460][__main__][INFO] - loss : 0.26510762769981655
[2024-03-26 07:56:19,614][__main__][INFO] - Epoch 17 with learning rate [0.0001]
[2024-03-26 10:06:39,738][__main__][INFO] - loss : 0.2632580618054603
[2024-03-26 10:17:23,706][__main__][INFO] - Epoch 18 with learning rate [0.0001]
[2024-03-26 12:27:54,357][__main__][INFO] - loss : 0.2617152291050864
[2024-03-26 12:38:49,432][__main__][INFO] - Epoch 19 with learning rate [0.0001]
[2024-03-26 14:49:32,657][__main__][INFO] - loss : 0.259896837311093
[2024-03-26 14:59:48,736][__main__][INFO] - Epoch 20 with learning rate [0.0001]
[2024-03-26 17:10:39,149][__main__][INFO] - loss : 0.25850626153643874
[2024-03-26 17:21:06,512][__main__][INFO] - Epoch 21 with learning rate [0.0001]
[2024-03-26 19:33:47,989][__main__][INFO] - loss : 0.25661532731617204
[2024-03-26 19:44:31,776][__main__][INFO] - Epoch 22 with learning rate [0.0001]
[2024-03-26 22:00:08,310][__main__][INFO] - loss : 0.25527636287169136
[2024-03-26 22:10:45,653][__main__][INFO] - Epoch 23 with learning rate [0.0001]
[2024-03-27 00:27:10,738][__main__][INFO] - loss : 0.25399190952605144
[2024-03-27 00:38:35,284][__main__][INFO] - Epoch 24 with learning rate [0.0001]
[2024-03-27 03:05:40,025][__main__][INFO] - loss : 0.25262153548892297
[2024-03-27 03:17:31,640][__main__][INFO] - Train time [hour] : 59.68981562011772 h
[2024-03-27 03:17:31,640][__main__][INFO] - Results saved in /root/LenslessPiCam/outputs/2024-03-24/15-35-50