Error: C:\Users\Dell\AI\pythonProject\.venv\Scripts\python.exe C:\Users\Dell\AI\pythonProject\!EridAI\VLM\main.py 
2024-10-06 17:46:28.205903: I tensorflow/core/util/port.cc:153] oneDNN custom operations are on. You may see slightly different numerical results due to floating-point round-off errors from different computation orders. To turn them off, set the environment variable `TF_ENABLE_ONEDNN_OPTS=0`.
2024-10-06 17:46:30.403259: I tensorflow/core/util/port.cc:153] oneDNN custom operations are on. You may see slightly different numerical results due to floating-point round-off errors from different computation orders. To turn them off, set the environment variable `TF_ENABLE_ONEDNN_OPTS=0`.
Default image_tensor shape: torch.Size([3, 224, 224])
Map (num_proc=4):   0%|          | 1/2434 [00:13<9:25:23, 13.94s/ examples]Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Map (num_proc=4):   0%|          | 12/2434 [00:14<30:41,  1.32 examples/s]Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Map (num_proc=4):   1%|          | 30/2434 [00:14<07:47,  5.14 examples/s]Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Map (num_proc=4):   2%|▏         | 37/2434 [00:15<05:53,  6.79 examples/s]Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Map (num_proc=4):   2%|▏         | 45/2434 [00:15<04:04,  9.78 examples/s]Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Map (num_proc=4):   2%|▏         | 54/2434 [00:15<02:48, 14.16 examples/s]Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Map (num_proc=4):   3%|▎         | 62/2434 [00:15<02:09, 18.25 examples/s]Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])Processed image_tensor shape: torch.Size([3, 224, 224])

Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Map (num_proc=4):   3%|▎         | 85/2434 [00:15<01:03, 37.02 examples/s]Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Map (num_proc=4):   4%|▍         | 104/2434 [00:15<00:44, 52.26 examples/s]Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Map (num_proc=4):   5%|▍         | 115/2434 [00:16<00:45, 50.50 examples/s]Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Map (num_proc=4):   5%|▌         | 127/2434 [00:16<00:39, 58.76 examples/s]Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Map (num_proc=4):   6%|▌         | 144/2434 [00:16<00:30, 73.91 examples/s]Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])

Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Map (num_proc=4):  10%|▉         | 239/2434 [00:17<00:20, 106.90 examples/s]Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Map (num_proc=4):  19%|█▉        | 471/2434 [00:19<00:14, 132.20 examples/s]Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])Processed image_tensor shape: torch.Size([3, 224, 224])

Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])Processed image_tensor shape: torch.Size([3, 224, 224])

Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])Processed image_tensor shape: torch.Size([3, 224, 224])

Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Map (num_proc=4):  26%|██▌       | 621/2434 [00:20<00:13, 134.48 examples/s]Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])Processed image_tensor shape: torch.Size([3, 224, 224])

Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])Processed image_tensor shape: torch.Size([3, 224, 224])


Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])Processed image_tensor shape: torch.Size([3, 224, 224])


Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])Processed image_tensor shape: torch.Size([3, 224, 224])

Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])

Map (num_proc=4):  38%|███▊      | 917/2434 [00:22<00:11, 137.36 examples/s]Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])Processed image_tensor shape: torch.Size([3, 224, 224])


Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])Processed image_tensor shape: torch.Size([3, 224, 224])

Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])Processed image_tensor shape: torch.Size([3, 224, 224])Processed image_tensor shape: torch.Size([3, 224, 224])


Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])

Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Map (num_proc=4):  50%|█████     | 1225/2434 [00:23<00:06, 186.26 examples/s]Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])Processed image_tensor shape: torch.Size([3, 224, 224])

Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])Processed image_tensor shape: torch.Size([3, 224, 224])Processed image_tensor shape: torch.Size([3, 224, 224])


Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])

Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])Processed image_tensor shape: torch.Size([3, 224, 224])Processed image_tensor shape: torch.Size([3, 224, 224])


Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])Processed image_tensor shape: torch.Size([3, 224, 224])Processed image_tensor shape: torch.Size([3, 224, 224])


Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Map (num_proc=4):  83%|████████▎ | 2014/2434 [00:28<00:02, 155.88 examples/s]Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Map (num_proc=4):  84%|████████▎ | 2033/2434 [00:28<00:03, 104.82 examples/s]Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Map (num_proc=4):  84%|████████▍ | 2052/2434 [00:28<00:03, 117.79 examples/s]Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Map (num_proc=4):  85%|████████▌ | 2076/2434 [00:29<00:02, 135.72 examples/s]Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Map (num_proc=4):  86%|████████▌ | 2096/2434 [00:29<00:02, 138.73 examples/s]Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Map (num_proc=4):  87%|████████▋ | 2123/2434 [00:29<00:01, 162.45 examples/s]Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Map (num_proc=4):  88%|████████▊ | 2143/2434 [00:29<00:01, 165.05 examples/s]Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Map (num_proc=4):  89%|████████▉ | 2161/2434 [00:29<00:01, 166.60 examples/s]Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Map (num_proc=4):  90%|████████▉ | 2180/2434 [00:29<00:01, 151.69 examples/s]Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Map (num_proc=4):  91%|█████████ | 2204/2434 [00:29<00:01, 164.38 examples/s]Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Map (num_proc=4):  91%|█████████▏| 2223/2434 [00:29<00:01, 165.36 examples/s]Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Map (num_proc=4):  92%|█████████▏| 2244/2434 [00:30<00:01, 155.34 examples/s]Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Map (num_proc=4):  96%|█████████▌| 2326/2434 [00:30<00:00, 111.13 examples/s]Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Map (num_proc=4):  99%|█████████▊| 2400/2434 [00:31<00:00, 77.08 examples/s]Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Processed image_tensor shape: torch.Size([3, 224, 224])
Map (num_proc=4): 100%|██████████| 2434/2434 [00:38<00:00, 63.41 examples/s]
Error: 'image' at index 0 is not a Tensor. Type: <class 'list'>
Error: 'image' at index 0 is not a Tensor. Type: <class 'list'>
Error: 'image' at index 0 is not a Tensor. Type: <class 'list'>
Error: 'image' at index 0 is not a Tensor. Type: <class 'list'>
Traceback (most recent call last):
  File "C:\Users\Dell\AI\pythonProject\!EridAI\VLM\main.py", line 242, in <module>
    main()
  File "C:\Users\Dell\AI\pythonProject\!EridAI\VLM\main.py", line 160, in main
    for batch_idx, batch in enumerate(dataloader):
                            ^^^^^^^^^^^^^^^^^^^^^
  File "C:\Users\Dell\AI\pythonProject\.venv\Lib\site-packages\torch\utils\data\dataloader.py", line 631, in __next__
    data = self._next_data()
           ^^^^^^^^^^^^^^^^^
  File "C:\Users\Dell\AI\pythonProject\.venv\Lib\site-packages\torch\utils\data\dataloader.py", line 1346, in _next_data
    return self._process_data(data)
           ^^^^^^^^^^^^^^^^^^^^^^^^
  File "C:\Users\Dell\AI\pythonProject\.venv\Lib\site-packages\torch\utils\data\dataloader.py", line 1372, in _process_data
    data.reraise()
  File "C:\Users\Dell\AI\pythonProject\.venv\Lib\site-packages\torch\_utils.py", line 705, in reraise
    raise exception
TypeError: Caught TypeError in DataLoader worker process 0.
Original Traceback (most recent call last):
  File "C:\Users\Dell\AI\pythonProject\.venv\Lib\site-packages\torch\utils\data\_utils\worker.py", line 308, in _worker_loop
    data = fetcher.fetch(index)  # type: ignore[possibly-undefined]
           ^^^^^^^^^^^^^^^^^^^^
  File "C:\Users\Dell\AI\pythonProject\.venv\Lib\site-packages\torch\utils\data\_utils\fetch.py", line 54, in fetch
    return self.collate_fn(data)
           ^^^^^^^^^^^^^^^^^^^^^
  File "C:\Users\Dell\AI\pythonProject\!EridAI\VLM\main.py", line 109, in collate_fn
    raise TypeError(f"Expected Tensor but got {type(img)} at index {idx}")
TypeError: Expected Tensor but got <class 'list'> at index 0


Process finished with exit code 1
 Code: import torch
from torch import nn
from torch.utils.data import DataLoader
from transformers import GPT2Tokenizer, GPT2LMHeadModel, CLIPProcessor, CLIPModel
from datasets import load_dataset
from PIL import Image
import numpy as np

class MultimodalGPT2(nn.Module):
    def __init__(self, gpt2_model_name, clip_model_name):
        super(MultimodalGPT2, self).__init__()
        self.clip = CLIPModel.from_pretrained(clip_model_name)
        self.gpt2 = GPT2LMHeadModel.from_pretrained(gpt2_model_name)
        self.adapter = nn.Linear(self.clip.config.projection_dim, self.gpt2.config.n_embd)

    def forward(self, input_ids, attention_mask, image=None):
        if image is not None:
            image_features = self.clip.get_image_features(pixel_values=image)
            adapter_features = self.adapter(image_features)
            adapter_features = adapter_features.unsqueeze(1).expand(-1, input_ids.size(1), -1)
            token_embeddings = self.gpt2.transformer.wte(input_ids)
            inputs_embeds = token_embeddings + adapter_features
            outputs = self.gpt2(
                inputs_embeds=inputs_embeds,
                attention_mask=attention_mask,
                labels=input_ids
            )
        else:
            outputs = self.gpt2(
                input_ids=input_ids,
                attention_mask=attention_mask,
                labels=input_ids
            )
        return outputs

def preprocess_data(example, processor, tokenizer, default_image):
    images = example.get('images', [])
    if images:
        try:
            image = images[0]

            # Ensure 'image' is not a list
            if isinstance(image, list):
                image = image[0]
                print(f"Warning: 'image' is a list, taking the first element.")

            # Convert to PIL Image if it's a path or ensure it's in RGB
            if isinstance(image, Image.Image):
                image = image.convert('RGB')
            else:
                image = Image.open(image).convert('RGB')

            # Process image with CLIPProcessor
            # Always pass a list of one image to ensure consistent processing
            encoding = processor(images=[image], return_tensors="pt")
            image_tensor = encoding['pixel_values']

            # Ensure 'pixel_values' is a tensor of shape (1, C, H, W)
            if isinstance(image_tensor, torch.Tensor):
                # Remove the batch dimension
                image_tensor = image_tensor.squeeze(0)  # Shape: (C, H, W)
                print(f"Processed image_tensor shape: {image_tensor.shape}")
            else:
                print(f"Warning: 'pixel_values' is not a Tensor. Type: {type(image_tensor)}. Converting to Tensor.")
                image_tensor = torch.tensor(image_tensor)

            # Validate the dimensions of 'image_tensor'
            if image_tensor.dim() != 3:
                print(f"Warning: 'image_tensor' has incorrect dimensions: {image_tensor.dim()}. Using default image.")
                image_tensor = default_image

        except Exception as e:
            print(f"Failed to process image: {e}. Using default image.")
            image_tensor = default_image
    else:
        image_tensor = default_image

    # Combine 'user' and 'assistant' texts
    text = (example.get('user', '') + " " + example.get('assistant', '')).strip()
    encoding = tokenizer(
        text, truncation=True, padding='max_length', max_length=128
    )

    # Debugging: Check types and shapes
    if not isinstance(image_tensor, torch.Tensor):
        print(f"Warning: 'image_tensor' is not a Tensor. Type: {type(image_tensor)}")
    elif image_tensor.dim() != 3:
        print(f"Warning: 'image_tensor' has incorrect dimensions. Shape: {image_tensor.shape}")

    return {
        'input_ids': encoding['input_ids'],
        'attention_mask': encoding['attention_mask'],
        'image': image_tensor
    }

def collate_fn(batch):
    # Explicitly specify dtype for input_ids and attention_mask
    input_ids = torch.tensor([item['input_ids'] for item in batch], dtype=torch.long)
    attention_mask = torch.tensor([item['attention_mask'] for item in batch], dtype=torch.long)

    images = []
    for idx, item in enumerate(batch):
        img = item['image']
        if img is not None:
            if isinstance(img, torch.Tensor):
                images.append(img)
            else:
                print(f"Error: 'image' at index {idx} is not a Tensor. Type: {type(img)}")
                raise TypeError(f"Expected Tensor but got {type(img)} at index {idx}")
        else:
            print(f"Error: 'image' at index {idx} is None.")
            raise ValueError(f"'Image' at index {idx} is None.")

    try:
        images = torch.stack(images)  # Now shape: (batch_size, C, H, W)
    except Exception as e:
        print(f"Failed to stack images: {e}")
        raise e

    return {'input_ids': input_ids, 'attention_mask': attention_mask, 'image': images}

def main():
    device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
    dataset = load_dataset('HuggingFaceM4/the_cauldron', 'ai2d', split='train')

    tokenizer = GPT2Tokenizer.from_pretrained('gpt2')
    tokenizer.pad_token = tokenizer.eos_token

    processor = CLIPProcessor.from_pretrained('openai/clip-vit-base-patch32')

    # Define a default image (e.g., a black image)
    default_pil_image = Image.fromarray(np.uint8(np.zeros((224, 224, 3))))
    encoding = processor(images=[default_pil_image], return_tensors="pt")
    default_image = encoding['pixel_values'].squeeze(0)  # Shape: (C, H, W)
    print(f"Default image_tensor shape: {default_image.shape}")

    # Preprocess the dataset with the default image
    processed_dataset = dataset.map(
        lambda x: preprocess_data(x, processor, tokenizer, default_image),
        remove_columns=dataset.column_names,
        num_proc=4  # Utilize multiple processes if possible
    )

    # Create DataLoader with enhanced collate_fn
    dataloader = DataLoader(
        processed_dataset,
        batch_size=8,
        shuffle=True,
        collate_fn=collate_fn,
        num_workers=4,  # Adjust based on your CPU cores
        pin_memory=True  # Speeds up data transfer to GPU
    )

    model = MultimodalGPT2('gpt2', 'openai/clip-vit-base-patch32').to(device)

    optimizer = torch.optim.AdamW(model.parameters(), lr=5e-5)  # Use PyTorch's AdamW

    model.train()
    for epoch in range(1):
        for batch_idx, batch in enumerate(dataloader):
            input_ids = batch['input_ids'].to(device)
            attention_mask = batch['attention_mask'].to(device)
            images = batch['image'].to(device)  # Ensured to have the same batch size

            outputs = model(input_ids=input_ids, attention_mask=attention_mask, image=images)
            loss = outputs.loss
            loss.backward()
            optimizer.step()
            optimizer.zero_grad()
            print(f"Epoch {epoch + 1}, Batch {batch_idx + 1}: Loss = {loss.item()}")

    torch.save(model.state_dict(), 'multimodal_gpt2.pth')
    model.eval()

    def generate(prompt_text=None, image_path=None):
        if prompt_text is not None:
            input_ids = tokenizer.encode(prompt_text, return_tensors='pt').to(device)
            attention_mask = torch.ones_like(input_ids).to(device)
        else:
            input_ids = torch.tensor([[tokenizer.eos_token_id]], dtype=torch.long).to(device)
            attention_mask = torch.ones_like(input_ids).to(device)

        if image_path is not None:
            try:
                image = Image.open(image_path).convert('RGB')
                encoding = processor(images=[image], return_tensors="pt")
                image_tensor = encoding['pixel_values']
                if isinstance(image_tensor, torch.Tensor):
                    image_tensor = image_tensor.squeeze(0)  # Shape: (C, H, W)
                else:
                    print(f"Warning: 'pixel_values' is not a Tensor. Type: {type(image_tensor)}. Converting to Tensor.")
                    image_tensor = torch.tensor(image_tensor)
                image_tensor = image_tensor.to(device)
                image_tensor = image_tensor.unsqueeze(0)  # Shape: (1, C, H, W)
                print(f"Generation using image, image_tensor shape: {image_tensor.shape}")
            except Exception as e:
                print(f"Error processing image: {e}. Using default image.")
                image_tensor = default_image.unsqueeze(0).to(device)
        else:
            # Use default image if no image_path is provided
            image_tensor = default_image.unsqueeze(0).to(device)

        with torch.no_grad():
            if image_tensor is not None:
                # Expand image to match batch size if necessary
                if image_tensor.size(0) != input_ids.size(0):
                    image_tensor = image_tensor.expand(input_ids.size(0), -1, -1, -1)
                image_features = model.clip.get_image_features(pixel_values=image_tensor)
                adapter_features = model.adapter(image_features)  # Shape: (batch_size, projection_dim)
                # Expand adapter_features to match sequence length
                adapter_features = adapter_features.unsqueeze(1).expand(-1, input_ids.size(1), -1)
                # Get token embeddings
                token_embeddings = model.gpt2.transformer.wte(input_ids)  # LongTensor
                # Combine token embeddings with adapter features
                inputs_embeds = token_embeddings + adapter_features
                # Generate with inputs_embeds
                outputs = model.gpt2.generate(
                    inputs_embeds=inputs_embeds,
                    attention_mask=attention_mask,
                    max_length=50
                )
            else:
                outputs = model.gpt2.generate(
                    input_ids=input_ids,
                    attention_mask=attention_mask,
                    max_length=50
                )

        return tokenizer.decode(outputs[0], skip_special_tokens=True)

    # Examples of generation
    print(generate(
        prompt_text="Question: What do respiration and combustion give out? Choices: A. Oxygen B. Carbon dioxide C. Nitrogen D. Heat Answer with the letter."
    ))
    print(generate(image_path='example.jpg'))
    print(generate(
        prompt_text="Question: What do respiration and combustion give out? Choices: A. Oxygen B. Carbon dioxide C. Nitrogen D. Heat Answer with the letter.",
        image_path='example.jpg'
    ))

if __name__ == "__main__":
    main()