| 11/26/2024 08:17:24 - INFO - __main__ - Distributed environment: MULTI_GPU Backend: nccl | |
| Num processes: 3 | |
| Process index: 0 | |
| Local process index: 0 | |
| Device: cuda:0 | |
| Mixed precision type: no | |
| 11/26/2024 08:17:38 - INFO - __main__ - Initializing the InstructPix2Pix UNet from the pretrained UNet. | |
| 11/26/2024 08:18:08 - INFO - __main__ - ***** Running training ***** | |
| 11/26/2024 08:18:08 - INFO - __main__ - Num examples = 108093 | |
| 11/26/2024 08:18:08 - INFO - __main__ - Num Epochs = 1 | |
| 11/26/2024 08:18:08 - INFO - __main__ - Instantaneous batch size per device = 8 | |
| 11/26/2024 08:18:08 - INFO - __main__ - Total train batch size (w. parallel, distributed & accumulation) = 48 | |
| 11/26/2024 08:18:08 - INFO - __main__ - Gradient Accumulation steps = 2 | |
| 11/26/2024 08:18:08 - INFO - __main__ - Total optimization steps = 100 | |