Without expanding dimension the following error comes up in while executing code block 33: example = train_dataset[0]
for k,v in example.items():
print(k,v.shape)
ValueError Traceback (most recent call last)
in ()
----> 1 example = train_dataset[0]
2 for k,v in example.items():
3 print(k,v.shape)
6 frames
/usr/local/lib/python3.10/dist-packages/transformers/image_utils.py in infer_channel_dimension_format(image, num_channels)
198 first_dim, last_dim = 1, 3
199 else:
--> 200 raise ValueError(f"Unsupported number of image dimensions: {image.ndim}")
201
202 if image.shape[first_dim] in num_channels:
ValueError: Unsupported number of image dimensions: 2
Without expanding dimension the following error comes up in while executing code block 33: example = train_dataset[0] for k,v in example.items(): print(k,v.shape)
ValueError Traceback (most recent call last)