Open prvnsingh opened 2 months ago
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'
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'