| [2024-12-07 16:13:58,819][__main__][INFO] - Using GPU for training. Main device : cuda:0 | |
| [2024-12-07 16:17:25,294][__main__][INFO] - PSF shape : torch.Size([1, 380, 507, 3]) | |
| [2024-12-07 16:17:25,316][__main__][INFO] - PSF min : 2.521084586456721e-18 | |
| [2024-12-07 16:17:25,320][__main__][INFO] - PSF max : 0.022293003275990486 | |
| [2024-12-07 16:17:25,320][__main__][INFO] - PSF dtype : torch.float32 | |
| [2024-12-07 16:17:25,343][__main__][INFO] - PSF norm : 1.0000226497650146 | |
| [2024-12-07 16:17:25,343][__main__][INFO] - Reconstruction a few images with ADMM... | |
| [2024-12-07 16:17:25,897][__main__][INFO] - Cropped shape : (200, 266, 3) | |
| [2024-12-07 16:17:27,811][__main__][INFO] - Train test size : 21250 | |
| [2024-12-07 16:17:27,811][__main__][INFO] - Test test size : 3750 | |
| [2024-12-07 16:17:27,922][__main__][INFO] - Training model with 8102338 parameters | |
| [2024-12-07 16:17:27,923][__main__][INFO] - -- Pre-process model with 4051168 parameters | |
| [2024-12-07 16:17:27,923][__main__][INFO] - -- Post-process model with 4051168 parameters | |
| [2024-12-07 16:17:27,923][__main__][INFO] - Setup time : 0.11219954490661621 s | |
| [2024-12-07 16:17:27,923][__main__][INFO] - PSF shape : torch.Size([1, 380, 507, 3]) | |
| [2024-12-07 16:17:27,923][__main__][INFO] - Results saved in /root/LenslessPiCam/outputs/2024-12-07/16-13-53 | |
| [2024-12-07 16:23:16,105][__main__][INFO] - Epoch 0 with learning rate [0.0001] | |
| [2024-12-07 17:37:32,347][__main__][INFO] - loss : 0.7107390846973657 | |
| [2024-12-07 17:43:20,341][__main__][INFO] - Epoch 1 with learning rate [0.0001] | |
| [2024-12-07 19:04:46,526][__main__][INFO] - loss : 0.6968980832623166 | |
| [2024-12-07 19:10:52,538][__main__][INFO] - Epoch 2 with learning rate [0.0001] | |
| [2024-12-07 20:58:29,519][__main__][INFO] - loss : 0.6805742324778454 | |
| [2024-12-07 21:04:32,810][__main__][INFO] - Epoch 3 with learning rate [0.0001] | |
| [2024-12-07 22:19:25,715][__main__][INFO] - loss : 0.67169693207171 | |
| [2024-12-07 22:25:29,029][__main__][INFO] - Epoch 4 with learning rate [0.0001] | |
| [2024-12-07 23:40:26,684][__main__][INFO] - loss : 0.66704060595834 | |
| [2024-12-07 23:46:31,359][__main__][INFO] - Epoch 5 with learning rate [0.0001] | |
| [2024-12-08 01:00:59,223][__main__][INFO] - loss : 0.6633480155892065 | |
| [2024-12-08 01:06:56,688][__main__][INFO] - Epoch 6 with learning rate [0.0001] | |
| [2024-12-08 02:21:24,082][__main__][INFO] - loss : 0.6607009546132525 | |
| [2024-12-08 02:27:23,310][__main__][INFO] - Epoch 7 with learning rate [0.0001] | |
| [2024-12-08 03:41:42,302][__main__][INFO] - loss : 0.6572266050466979 | |
| [2024-12-08 03:47:40,988][__main__][INFO] - Epoch 8 with learning rate [0.0001] | |
| [2024-12-08 05:02:04,599][__main__][INFO] - loss : 0.6525141091506486 | |
| [2024-12-08 05:08:00,998][__main__][INFO] - Epoch 9 with learning rate [0.0001] | |
| [2024-12-08 06:22:24,081][__main__][INFO] - loss : 0.6486295952750731 | |
| [2024-12-08 06:28:18,986][__main__][INFO] - Epoch 10 with learning rate [0.0001] | |
| [2024-12-08 07:42:38,398][__main__][INFO] - loss : 0.645721460323247 | |
| [2024-12-08 07:48:34,454][__main__][INFO] - Epoch 11 with learning rate [0.0001] | |
| [2024-12-08 09:02:59,048][__main__][INFO] - loss : 0.6434066692258869 | |
| [2024-12-08 09:08:52,956][__main__][INFO] - Epoch 12 with learning rate [0.0001] | |
| [2024-12-08 10:23:12,625][__main__][INFO] - loss : 0.6414004320448103 | |
| [2024-12-08 10:29:06,850][__main__][INFO] - Epoch 13 with learning rate [0.0001] | |
| [2024-12-08 11:43:19,109][__main__][INFO] - loss : 0.6393773910152512 | |
| [2024-12-08 11:49:12,297][__main__][INFO] - Epoch 14 with learning rate [0.0001] | |
| [2024-12-08 13:03:28,302][__main__][INFO] - loss : 0.6381175140977333 | |
| [2024-12-08 13:09:23,885][__main__][INFO] - Epoch 15 with learning rate [0.0001] | |
| [2024-12-08 14:23:43,439][__main__][INFO] - loss : 0.636914866788003 | |
| [2024-12-08 14:29:40,724][__main__][INFO] - Epoch 16 with learning rate [0.0001] | |
| [2024-12-08 15:44:03,273][__main__][INFO] - loss : 0.6358224534672032 | |
| [2024-12-08 15:49:59,673][__main__][INFO] - Epoch 17 with learning rate [0.0001] | |
| [2024-12-08 17:04:22,148][__main__][INFO] - loss : 0.6347603342604038 | |
| [2024-12-08 17:10:17,928][__main__][INFO] - Epoch 18 with learning rate [0.0001] | |
| [2024-12-08 18:24:32,689][__main__][INFO] - loss : 0.6337339697630417 | |
| [2024-12-08 18:30:24,673][__main__][INFO] - Epoch 19 with learning rate [0.0001] | |
| [2024-12-08 19:44:41,323][__main__][INFO] - loss : 0.6326317376904695 | |
| [2024-12-08 19:50:34,278][__main__][INFO] - Epoch 20 with learning rate [0.0001] | |
| [2024-12-08 21:04:54,703][__main__][INFO] - loss : 0.6319631520646836 | |
| [2024-12-08 21:10:48,058][__main__][INFO] - Epoch 21 with learning rate [0.0001] | |
| [2024-12-08 22:25:04,632][__main__][INFO] - loss : 0.6308716587678801 | |
| [2024-12-08 22:30:59,968][__main__][INFO] - Epoch 22 with learning rate [0.0001] | |
| [2024-12-08 23:45:10,197][__main__][INFO] - loss : 0.630242119131848 | |
| [2024-12-08 23:51:04,096][__main__][INFO] - Epoch 23 with learning rate [0.0001] | |
| [2024-12-09 01:05:08,751][__main__][INFO] - loss : 0.6292254139132692 | |
| [2024-12-09 01:10:53,964][__main__][INFO] - Epoch 24 with learning rate [0.0001] | |
| [2024-12-09 02:24:46,212][__main__][INFO] - loss : 0.6285865292157822 | |
| [2024-12-09 02:30:31,260][__main__][INFO] - Train time [hour] : 34.21720594253805 h | |
| [2024-12-09 02:30:31,260][__main__][INFO] - Results saved in /root/LenslessPiCam/outputs/2024-12-07/16-13-53 | |