
debug笔记
沃·夏澈德
今天的明天是后天的昨天。----茨鲍勒·程德
展开
-
pytorch Debug笔记:TypeError: __init__() got an unexpected keyword argument ‘data_tensor‘
报错代码torch_dataset = Data.TensorDataset(data_tensor=data, target_tensor=label)报错TypeError: __init__() got an unexpected keyword argument 'data_tensor'原因:时代变了,新版把输入变成了可变参数,直接输入即可正确写法torch_dataset = Data.TensorDataset(data, label)...原创 2021-10-24 20:57:29 · 9854 阅读 · 0 评论 -
pytorch debug笔记:当训练过程中loss忽然变为 nan
原因有很多,此处给出是 除0 产生的问题。但排查思路是相通的。方法较原始,但有效。首先在网络的每个步骤后添加一个if,看是否产生 nandef forward(self, source): source = self.blocks(source) if torch.any(torch.isnan(source)): print(1) source = source.view(source.size()[0], -1) #原创 2021-10-27 16:07:51 · 1238 阅读 · 2 评论