I have made a custom dataset to read images as well as texts together , however I am getting this error :
RuntimeError: number of dims don\'t match in p