[2024-05-06 11:52:03,388][__main__][INFO] - Using GPU for training. Main device : cuda:1
[2024-05-06 11:52:03,388][__main__][INFO] - Using multiple GPUs : [1, 2, 3]
[2024-05-06 11:52:12,484][__main__][INFO] - PSF shape : torch.Size([1, 380, 507, 3])
[2024-05-06 11:52:12,505][__main__][INFO] - PSF min : 3.7760526956527246e-17
[2024-05-06 11:52:12,511][__main__][INFO] - PSF max : 0.02279210090637207
[2024-05-06 11:52:12,511][__main__][INFO] - PSF dtype : torch.float32
[2024-05-06 11:52:12,527][__main__][INFO] - PSF norm : 1.0000208616256714
[2024-05-06 11:52:12,527][__main__][INFO] - Reconstruction a few images with ADMM...
[2024-05-06 11:52:13,112][__main__][INFO] - Cropped shape :  (200, 266, 3)
[2024-05-06 11:52:13,174][matplotlib.image][WARNING] - Clipping input data to the valid range for imshow with RGB data ([0..1] for floats or [0..255] for integers).
[2024-05-06 11:52:13,662][matplotlib.image][WARNING] - Clipping input data to the valid range for imshow with RGB data ([0..1] for floats or [0..255] for integers).
[2024-05-06 11:52:14,120][matplotlib.image][WARNING] - Clipping input data to the valid range for imshow with RGB data ([0..1] for floats or [0..255] for integers).
[2024-05-06 11:52:14,576][matplotlib.image][WARNING] - Clipping input data to the valid range for imshow with RGB data ([0..1] for floats or [0..255] for integers).
[2024-05-06 11:52:15,024][matplotlib.image][WARNING] - Clipping input data to the valid range for imshow with RGB data ([0..1] for floats or [0..255] for integers).
[2024-05-06 11:52:15,268][__main__][INFO] - Train test size : 21250
[2024-05-06 11:52:15,268][__main__][INFO] - Test test size : 3750
[2024-05-06 11:52:15,289][__main__][INFO] - Training model with 40 parameters
[2024-05-06 11:52:15,289][__main__][INFO] - Setup time : 0.021241426467895508 s
[2024-05-06 11:52:15,290][__main__][INFO] - PSF shape : torch.Size([1, 380, 507, 3])
[2024-05-06 11:52:15,290][__main__][INFO] - Results saved in /root/LenslessPiCam/outputs/2024-05-06/11-51-57
[2024-05-06 11:57:37,280][__main__][INFO] - Epoch 0 with learning rate [0.0001]
[2024-05-06 13:00:44,923][__main__][INFO] - loss : 0.832464786610675
[2024-05-06 13:05:58,342][__main__][INFO] - Epoch 1 with learning rate [0.0001]
[2024-05-06 14:09:05,178][__main__][INFO] - loss : 0.816342409196018
[2024-05-06 14:14:25,046][__main__][INFO] - Epoch 2 with learning rate [0.0001]
[2024-05-06 15:17:42,139][__main__][INFO] - loss : 0.8114292586946963
[2024-05-06 15:22:59,249][__main__][INFO] - Epoch 3 with learning rate [0.0001]
[2024-05-06 16:25:51,189][__main__][INFO] - loss : 0.8092869028025561
[2024-05-06 16:31:09,659][__main__][INFO] - Epoch 4 with learning rate [0.0001]
[2024-05-06 17:34:23,411][__main__][INFO] - loss : 0.8083697364937451
[2024-05-06 17:39:43,366][__main__][INFO] - Epoch 5 with learning rate [0.0001]
[2024-05-06 18:42:30,810][__main__][INFO] - loss : 0.8057006941258119
[2024-05-06 18:47:48,196][__main__][INFO] - Epoch 6 with learning rate [0.0001]
[2024-05-06 19:51:02,470][__main__][INFO] - loss : 0.8043194443251052
[2024-05-06 19:56:21,366][__main__][INFO] - Epoch 7 with learning rate [0.0001]
[2024-05-06 20:59:11,273][__main__][INFO] - loss : 0.8027254983729497
[2024-05-06 21:04:32,986][__main__][INFO] - Epoch 8 with learning rate [0.0001]
[2024-05-06 22:07:48,409][__main__][INFO] - loss : 0.7956307039475364
[2024-05-06 22:13:08,869][__main__][INFO] - Epoch 9 with learning rate [0.0001]
[2024-05-06 23:15:51,864][__main__][INFO] - loss : 0.7866104962057074
[2024-05-06 23:21:09,961][__main__][INFO] - Epoch 10 with learning rate [0.0001]
[2024-05-07 00:24:23,051][__main__][INFO] - loss : 0.7848451108486444
[2024-05-07 00:29:40,792][__main__][INFO] - Epoch 11 with learning rate [0.0001]
[2024-05-07 01:32:27,439][__main__][INFO] - loss : 0.7845522786923441
[2024-05-07 01:37:46,414][__main__][INFO] - Epoch 12 with learning rate [0.0001]
[2024-05-07 02:41:09,418][__main__][INFO] - loss : 0.7837262168321915
[2024-05-07 02:46:30,802][__main__][INFO] - Epoch 13 with learning rate [0.0001]
[2024-05-07 03:49:20,949][__main__][INFO] - loss : 0.7830422357069707
[2024-05-07 03:54:41,926][__main__][INFO] - Epoch 14 with learning rate [0.0001]
[2024-05-07 04:58:01,658][__main__][INFO] - loss : 0.7826564252858962
[2024-05-07 05:03:20,686][__main__][INFO] - Epoch 15 with learning rate [0.0001]
[2024-05-07 06:06:10,838][__main__][INFO] - loss : 0.7821302957034875
[2024-05-07 06:11:30,946][__main__][INFO] - Epoch 16 with learning rate [0.0001]
[2024-05-07 07:14:45,832][__main__][INFO] - loss : 0.7821178691615591
[2024-05-07 07:20:06,924][__main__][INFO] - Epoch 17 with learning rate [0.0001]
[2024-05-07 08:23:19,226][__main__][INFO] - loss : 0.7817506057022874
[2024-05-07 08:28:45,638][__main__][INFO] - Epoch 18 with learning rate [0.0001]
[2024-05-07 09:32:42,466][__main__][INFO] - loss : 0.7808720612961227
[2024-05-07 09:38:01,482][__main__][INFO] - Epoch 19 with learning rate [0.0001]
[2024-05-07 10:40:46,764][__main__][INFO] - loss : 0.780778784503287
[2024-05-07 10:46:05,054][__main__][INFO] - Epoch 20 with learning rate [0.0001]
[2024-05-07 11:49:14,372][__main__][INFO] - loss : 0.7804283967310728
[2024-05-07 11:54:34,150][__main__][INFO] - Epoch 21 with learning rate [0.0001]
[2024-05-07 12:57:22,371][__main__][INFO] - loss : 0.7799849033849331
[2024-05-07 13:02:43,496][__main__][INFO] - Epoch 22 with learning rate [0.0001]
[2024-05-07 14:06:02,688][__main__][INFO] - loss : 0.7798809090083368
[2024-05-07 14:11:21,462][__main__][INFO] - Epoch 23 with learning rate [0.0001]
[2024-05-07 15:14:08,597][__main__][INFO] - loss : 0.7796271378859825
[2024-05-07 15:19:26,673][__main__][INFO] - Epoch 24 with learning rate [0.0001]
[2024-05-07 16:22:44,408][__main__][INFO] - loss : 0.7789510395507578
[2024-05-07 16:28:04,900][__main__][INFO] - Train time [hour] : 28.596722996168666 h
[2024-05-07 16:28:04,901][__main__][INFO] - Results saved in /root/LenslessPiCam/outputs/2024-05-06/11-51-57