我们经常在深度学习训练模型的过程中,会需要我们进行一个debug的过程,主要去看看数据的维度、以及预测的结果的数据是否符合内心的预期。
比如会出现下面的问题:
注意:这个是模型训练的过程,主要会有下面的标志:grad_fn=<.......>,我们可以在下面debug的时候右键选择“Evaluate Expression”
接着选择输入你想要查看的数据类型:比如我要查看训练过程中的输出预测query_pred_seg(tensor)类型。我在后面加上.clone().detach().cpu().numpy(),就可以查看到我们想要的numpy数据内容了。
内容展示: