tensor(0.7428, grad_fn=<DivBackward0>) 取出数据

用模型训练计算loss的时候,loss的结果是:
tensor(0.7428, grad_fn=)
如果想绘图的话,需要单独将数据取出,取出的方法是x.item()
例如:

x = torch.tensor(0.8806, requires_grad=True)
print(x.item())

结果是这样的:

0.8805999755859375

不知道为什么会有数位的变化,路过的可否告知一下~

那么在训练的时候就这样写

 all_epoch_loss = []
 all_epoch_loss.append(epoch_mean_loss.item())  

接下来绘图的时候直接用这个list就可以了。

你可能感兴趣的:(python,jupyter,编辑器)