神经网络的一个报错,一般是BatchNormal2d (args*),可能原因是args参数错了,这里的参数应当是上一层的channel数,而不是特征数(batch*channel*w*h)。修改一下即可
RuntimeError: running_mean should contain 48 elements not 324480
最新推荐文章于 2024-05-18 13:01:50 发布
神经网络的一个报错,一般是BatchNormal2d (args*),可能原因是args参数错了,这里的参数应当是上一层的channel数,而不是特征数(batch*channel*w*h)。修改一下即可