I have a custom dataset as below:
class VQAdataset(Dataset): def __init__(self,img_data, que_data, ans_data, top_1000_answers): #self.csv_file=pd.read_c