原因分析:数据在cuda() ,需要将 float 型的tensor转化为numpy ,只能在cpu上进行
更改前的样子:
labels_=labels_.numpy().tolist()
更改后的样子哦
labels_=labels_.cpu().numpy().tolist()
2023-12-05 22:21:58
原因分析:数据在cuda() ,需要将 float 型的tensor转化为numpy ,只能在cpu上进行
更改前的样子:
labels_=labels_.numpy().tolist()
更改后的样子哦
labels_=labels_.cpu().numpy().tolist()