有点匪夷所思,错误解决是在 model 模块,

# 报错
logits = self.linear(pooled_output)

# 没问题了
logits = self.linear(pooled_output).view(batch_size, self.num_classes)

 

Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐