问题描述 训练时出现nan,有时出现在刚开始的时候,有时出现在训练中。 解决 debug,发现是BatchNorm时出现了问题。我的代码中有的batch size为1,这时 torch.var(data,dim=0) 会返回nan