ZexinYan/Medical-Report-Generation

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