Closed Clitost closed 4 years ago
在forward的x = x.squeeze()这里,如果使用者最后一个batch的大小刚好是1时,会被squeeze掉,这样输出标签的维度不匹配,建议修改。
在forward的x = x.squeeze()这里,如果使用者最后一个batch的大小刚好是1时,会被squeeze掉,这样输出标签的维度不匹配,建议修改。