The problem of dataloader
wlufy opened this issue · 0 comments
wlufy commented
targets = np.zeros((len(captions), max_sentence_num + 1, max_word_num))
prob = np.zeros((len(captions), max_sentence_num + 1))
When i read the dataset code, i found that the second dim of this two variable added to 1. I don't know why it's set up this way, could anyone help answer this? Thx