[2023-10-31 18:26:08,047][__main__][INFO] - Using GPU for training.
[2023-10-31 18:26:09,604][__main__][INFO] - Data shape :  torch.Size([1, 380, 507, 3])
[2023-10-31 18:26:09,605][__main__][INFO] - PSF shape : torch.Size([1, 380, 507, 3])
[2023-10-31 18:26:09,606][__main__][INFO] - PSF min : 0.0
[2023-10-31 18:26:09,606][__main__][INFO] - PSF max : 0.007229802664369345
[2023-10-31 18:26:09,606][__main__][INFO] - PSF dtype : torch.float32
[2023-10-31 18:26:09,606][__main__][INFO] - PSF norm : 1.0000030994415283
[2023-10-31 18:26:09,606][__main__][INFO] - Reconstruction a few images with ADMM...
[2023-10-31 18:26:09,813][__main__][INFO] - Cropped shape :  (262, 215, 3)
[2023-10-31 18:26:11,158][__main__][INFO] - Train test size : 22100
[2023-10-31 18:26:11,158][__main__][INFO] - Test test size : 3900
[2023-10-31 18:26:11,224][__main__][INFO] - Training model with 8102378 parameters
[2023-10-31 18:26:11,224][__main__][INFO] - Setup time : 0.06587958335876465 s
[2023-10-31 18:26:11,224][__main__][INFO] - PSF shape : torch.Size([1, 380, 507, 3])
[2023-10-31 18:26:11,224][__main__][INFO] - Results saved in /home/bezzam/LenslessPiCam/outputs/2023-10-31/18-26-08
[2023-10-31 18:31:52,736][__main__][INFO] - Epoch 0 with learning rate [0.0001]
[2023-10-31 19:54:54,906][__main__][INFO] - loss : 0.5285163230216329
[2023-10-31 20:00:33,117][__main__][INFO] - Epoch 1 with learning rate [0.0001]
[2023-10-31 21:23:32,001][__main__][INFO] - loss : 0.39819716731347704
[2023-10-31 21:29:10,559][__main__][INFO] - Epoch 2 with learning rate [0.0001]
[2023-10-31 22:52:06,869][__main__][INFO] - loss : 0.3741332387762381
[2023-10-31 22:57:45,399][__main__][INFO] - Epoch 3 with learning rate [0.0001]
[2023-11-01 00:20:40,450][__main__][INFO] - loss : 0.36009180308467154
[2023-11-01 00:26:18,484][__main__][INFO] - Epoch 4 with learning rate [0.0001]
[2023-11-01 01:49:14,604][__main__][INFO] - loss : 0.34870623931356204
[2023-11-01 01:54:52,941][__main__][INFO] - Epoch 5 with learning rate [0.0001]
[2023-11-01 03:17:47,591][__main__][INFO] - loss : 0.33998378451593336
[2023-11-01 03:23:25,749][__main__][INFO] - Epoch 6 with learning rate [0.0001]
[2023-11-01 04:46:20,243][__main__][INFO] - loss : 0.33219305791466447
[2023-11-01 04:51:58,034][__main__][INFO] - Epoch 7 with learning rate [0.0001]
[2023-11-01 06:14:49,947][__main__][INFO] - loss : 0.32572161962273943
[2023-11-01 06:20:27,767][__main__][INFO] - Epoch 8 with learning rate [0.0001]
[2023-11-01 07:43:19,044][__main__][INFO] - loss : 0.32021956704590565
[2023-11-01 07:48:56,760][__main__][INFO] - Epoch 9 with learning rate [0.0001]
[2023-11-01 09:11:50,538][__main__][INFO] - loss : 0.31551459585649383
[2023-11-01 09:17:28,242][__main__][INFO] - Epoch 10 with learning rate [0.0001]
[2023-11-01 10:40:19,710][__main__][INFO] - loss : 0.3114133763070564
[2023-11-01 10:45:57,252][__main__][INFO] - Epoch 11 with learning rate [0.0001]
[2023-11-01 12:08:51,959][__main__][INFO] - loss : 0.30743926291012635
[2023-11-01 12:14:31,061][__main__][INFO] - Epoch 12 with learning rate [0.0001]
[2023-11-01 13:38:05,838][__main__][INFO] - loss : 0.30434536739981527
[2023-11-01 13:43:52,014][__main__][INFO] - Epoch 13 with learning rate [0.0001]
[2023-11-01 15:07:47,408][__main__][INFO] - loss : 0.30100269020412834
[2023-11-01 15:13:32,818][__main__][INFO] - Epoch 14 with learning rate [0.0001]
[2023-11-01 16:37:30,761][__main__][INFO] - loss : 0.29866309750673564
[2023-11-01 16:43:19,606][__main__][INFO] - Epoch 15 with learning rate [0.0001]
[2023-11-01 18:07:14,244][__main__][INFO] - loss : 0.296072699254993
[2023-11-01 18:12:59,805][__main__][INFO] - Epoch 16 with learning rate [0.0001]
[2023-11-01 19:36:50,835][__main__][INFO] - loss : 0.293984055831961
[2023-11-01 19:42:36,309][__main__][INFO] - Epoch 17 with learning rate [0.0001]
[2023-11-01 21:06:27,254][__main__][INFO] - loss : 0.29159652836182537
[2023-11-01 21:12:12,620][__main__][INFO] - Epoch 18 with learning rate [0.0001]
[2023-11-01 22:36:03,774][__main__][INFO] - loss : 0.28966026354429436
[2023-11-01 22:41:48,432][__main__][INFO] - Epoch 19 with learning rate [0.0001]
[2023-11-02 00:05:34,970][__main__][INFO] - loss : 0.2875611148565611
[2023-11-02 00:11:19,938][__main__][INFO] - Epoch 20 with learning rate [0.0001]
[2023-11-02 01:35:09,108][__main__][INFO] - loss : 0.28574022267738697
[2023-11-02 01:40:54,218][__main__][INFO] - Epoch 21 with learning rate [0.0001]
[2023-11-02 03:04:42,701][__main__][INFO] - loss : 0.28423031353572886
[2023-11-02 03:10:26,777][__main__][INFO] - Epoch 22 with learning rate [0.0001]
[2023-11-02 04:34:17,500][__main__][INFO] - loss : 0.28256982792556523
[2023-11-02 04:40:02,421][__main__][INFO] - Epoch 23 with learning rate [0.0001]
[2023-11-02 06:03:51,317][__main__][INFO] - loss : 0.28112380514047697
[2023-11-02 06:09:36,003][__main__][INFO] - Epoch 24 with learning rate [0.0001]
[2023-11-02 07:33:26,372][__main__][INFO] - loss : 0.2795201335773206
[2023-11-02 07:39:11,053][__main__][INFO] - Train time : 133979.0553715229 s
[2023-11-02 07:39:11,053][__main__][INFO] - Results saved in /home/bezzam/LenslessPiCam/outputs/2023-10-31/18-26-08