TypeError: iteration over a 0-d tensor

模型训练报错:
TypeError: iteration over a 0-d tensor
解决:
total_loss+=loss
换成如下方式
total_loss+=loss.item()

你可能感兴趣的:(NLP,自然语言处理,数据挖掘)