| [2024-06-19 09:59:27,786][__main__][INFO] - Using GPU for training. Main device : cuda:0 | |
| [2024-06-19 09:59:27,787][__main__][INFO] - Using multiple GPUs : [0, 3] | |
| [2024-06-19 10:03:01,415][__main__][INFO] - PSF shape : torch.Size([1, 135, 240, 3]) | |
| [2024-06-19 10:03:01,432][__main__][INFO] - PSF min : 0.0 | |
| [2024-06-19 10:03:01,443][__main__][INFO] - PSF max : 0.12337537109851837 | |
| [2024-06-19 10:03:01,443][__main__][INFO] - PSF dtype : torch.float32 | |
| [2024-06-19 10:03:01,474][__main__][INFO] - PSF norm : 1.7320507764816284 | |
| [2024-06-19 10:03:01,474][__main__][INFO] - Reconstruction a few images with ADMM... | |
| [2024-06-19 10:03:02,229][__main__][INFO] - Cropped shape : (135, 240, 3) | |
| [2024-06-19 10:03:03,546][__main__][INFO] - Train test size : 24000 | |
| [2024-06-19 10:03:03,547][__main__][INFO] - Test test size : 999 | |
| [2024-06-19 10:03:03,632][__main__][INFO] - Training model with 8106693 parameters | |
| [2024-06-19 10:03:03,632][__main__][INFO] - Setup time : 0.08576440811157227 s | |
| [2024-06-19 10:03:03,633][__main__][INFO] - PSF shape : torch.Size([1, 135, 240, 1]) | |
| [2024-06-19 10:03:03,633][__main__][INFO] - Results saved in /root/LenslessPiCam/outputs/2024-06-19/09-59-22 | |
| [2024-06-19 10:04:00,697][__main__][INFO] - Epoch 0 with learning rate [0.0001] | |
| [2024-06-19 10:33:36,703][__main__][INFO] - loss : 0.4578864697466301 | |
| [2024-06-19 10:34:28,459][__main__][INFO] - Epoch 1 with learning rate [0.0001] | |
| [2024-06-19 11:04:28,497][__main__][INFO] - loss : 0.37156011185546695 | |
| [2024-06-19 11:05:20,430][__main__][INFO] - Epoch 2 with learning rate [0.0001] | |
| [2024-06-19 11:35:03,945][__main__][INFO] - loss : 0.32595192829519454 | |
| [2024-06-19 11:35:57,732][__main__][INFO] - Epoch 3 with learning rate [0.0001] | |
| [2024-06-19 12:05:47,091][__main__][INFO] - loss : 0.3018787100588286 | |
| [2024-06-19 12:06:38,482][__main__][INFO] - Epoch 4 with learning rate [0.0001] | |
| [2024-06-19 12:34:23,419][__main__][INFO] - loss : 0.28623140198489017 | |
| [2024-06-19 12:35:13,561][__main__][INFO] - Epoch 5 with learning rate [0.0001] | |
| [2024-06-19 13:02:41,679][__main__][INFO] - loss : 0.27335716788719105 | |
| [2024-06-19 13:03:31,158][__main__][INFO] - Epoch 6 with learning rate [0.0001] | |
| [2024-06-19 13:30:48,710][__main__][INFO] - loss : 0.26367658338198996 | |
| [2024-06-19 13:31:38,596][__main__][INFO] - Epoch 7 with learning rate [0.0001] | |
| [2024-06-19 13:59:01,883][__main__][INFO] - loss : 0.25540653997659724 | |
| [2024-06-19 13:59:51,912][__main__][INFO] - Epoch 8 with learning rate [0.0001] | |
| [2024-06-19 14:31:48,658][__main__][INFO] - loss : 0.24810290103157356 | |
| [2024-06-19 14:32:48,491][__main__][INFO] - Epoch 9 with learning rate [0.0001] | |
| [2024-06-19 15:13:32,577][__main__][INFO] - loss : 0.2431616224795581 | |
| [2024-06-19 15:14:34,245][__main__][INFO] - Epoch 10 with learning rate [0.0001] | |
| [2024-06-19 15:54:41,008][__main__][INFO] - loss : 0.23770997892568443 | |
| [2024-06-19 15:55:39,915][__main__][INFO] - Epoch 11 with learning rate [0.0001] | |
| [2024-06-19 16:36:22,005][__main__][INFO] - loss : 0.2338216504653295 | |
| [2024-06-19 16:37:21,294][__main__][INFO] - Epoch 12 with learning rate [0.0001] | |
| [2024-06-19 17:17:18,560][__main__][INFO] - loss : 0.22913359238455713 | |
| [2024-06-19 17:18:17,382][__main__][INFO] - Epoch 13 with learning rate [0.0001] | |
| [2024-06-19 17:58:59,122][__main__][INFO] - loss : 0.22463069409380368 | |
| [2024-06-19 17:59:57,881][__main__][INFO] - Epoch 14 with learning rate [0.0001] | |
| [2024-06-19 18:40:04,466][__main__][INFO] - loss : 0.22047558621317162 | |
| [2024-06-19 18:41:02,791][__main__][INFO] - Epoch 15 with learning rate [0.0001] | |
| [2024-06-19 19:21:37,902][__main__][INFO] - loss : 0.21713632159183385 | |
| [2024-06-19 19:22:37,399][__main__][INFO] - Epoch 16 with learning rate [0.0001] | |
| [2024-06-19 20:02:45,450][__main__][INFO] - loss : 0.21379765899976033 | |
| [2024-06-19 20:03:45,244][__main__][INFO] - Epoch 17 with learning rate [0.0001] | |
| [2024-06-19 20:44:24,304][__main__][INFO] - loss : 0.21089778315400085 | |
| [2024-06-19 20:45:24,958][__main__][INFO] - Epoch 18 with learning rate [0.0001] | |
| [2024-06-19 21:25:28,685][__main__][INFO] - loss : 0.20836715550348203 | |
| [2024-06-19 21:26:27,126][__main__][INFO] - Epoch 19 with learning rate [0.0001] | |
| [2024-06-19 22:07:12,466][__main__][INFO] - loss : 0.20587286516651515 | |
| [2024-06-19 22:08:13,438][__main__][INFO] - Epoch 20 with learning rate [0.0001] | |
| [2024-06-19 22:48:08,827][__main__][INFO] - loss : 0.2033102456592016 | |
| [2024-06-19 22:49:07,348][__main__][INFO] - Epoch 21 with learning rate [0.0001] | |
| [2024-06-19 23:29:53,772][__main__][INFO] - loss : 0.20088441715389468 | |
| [2024-06-19 23:30:52,443][__main__][INFO] - Epoch 22 with learning rate [0.0001] | |
| [2024-06-20 00:11:03,001][__main__][INFO] - loss : 0.19855512122685706 | |
| [2024-06-20 00:12:02,370][__main__][INFO] - Epoch 23 with learning rate [0.0001] | |
| [2024-06-20 00:52:41,740][__main__][INFO] - loss : 0.19672937769815296 | |
| [2024-06-20 00:53:40,836][__main__][INFO] - Epoch 24 with learning rate [0.0001] | |
| [2024-06-20 01:33:38,852][__main__][INFO] - loss : 0.1945796014716229 | |
| [2024-06-20 01:34:37,608][__main__][INFO] - Train time [hour] : 15.525681910845968 h | |
| [2024-06-20 01:34:37,609][__main__][INFO] - Results saved in /root/LenslessPiCam/outputs/2024-06-19/09-59-22 | |