07/10/2023 17:45:20 - INFO - __main__ - Distributed environment: MULTI_GPU  Backend: nccl
Num processes: 3
Process index: 1
Local process index: 1
Device: cuda:1
Mixed precision type: fp16
Resolving data files: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████| 229/229 [00:00<00:00, 30032.38it/s]
Downloading data files: 100%|██████████████████████████████████████████████████████████████████████████████████████████████████████| 229/229 [00:00<00:00, 36149.63it/s]
Downloading data files: 0it [00:00, ?it/s]
Extracting data files: 0it [00:00, ?it/s]
Downloading and preparing dataset imagefolder/default to /l/vision/v5/sragas/hf_models/imagefolder/default-2bcaa40744b68173/0.0.0/37fbb85cc714a338bea574ac6c7d0b5be5aff46c1862c1989b20e0771199e93f...
Dataset imagefolder downloaded and prepared to /l/vision/v5/sragas/hf_models/imagefolder/default-2bcaa40744b68173/0.0.0/37fbb85cc714a338bea574ac6c7d0b5be5aff46c1862c1989b20e0771199e93f. Subsequent calls will reuse this data.