vis-nlp / ChartInstruct

GNU General Public License v3.0
18 stars 1 forks source link

Error in FineTuning #5

Open prvnsingh opened 2 months ago

prvnsingh commented 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'