yangjianxin1 / ClipCap-Chinese

基于ClipCap的看图说话Image Caption模型

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

shift_logits

ASENNIU opened this issue · comments

作者你好,我是nlp初学者,请问为什么在计算交叉熵的时候要shift logits喃。
hift_logits = logits[..., args.prefix_len - 1:-1, :].contiguous().view(-1, logits.size(-1))