Error in FineTuning #5
Description
The processor does not return image sizes.
62 File "/home/ec2-user/anaconda3/envs/gcc/lib/python3.12/site-packages/torch/utils/data/_utils/fetch.py", line 54, in fetch
63 return self.collate_fn(data)
64 ^^^^^^^^^^^^^^^^^^^^^
65 File "/home/ec2-user/SageMaker/fineTune_chart_new.py", line 241, in train_collate_fn
66 image_sizes = batch["image_sizes"]
67 ~~~~~^^^^^^^^^^^^^^^
68 File "/home/ec2-user/anaconda3/envs/gcc/lib/python3.12/site-packages/transformers/feature_extraction_utils.py", line 87, in getitem
69 return self.data[item]
70 ~~~~~~~~~^^^^^^
71 KeyError: 'image_sizes'