pytorch模型中添加batchnorm过程中出现错误RuntimeError: running_mean should contain 1 elements not 512
,错误来自于nn.BatchNorm1d(N),其中N=channels
解决办法:
将输入重新排列
video = video.view(-1, video.shape[-1])
pytorch模型中添加batchnorm过程中出现错误RuntimeError: running_mean should contain 1 elements not 512
,错误来自于nn.BatchNorm1d(N),其中N=channels
解决办法:
将输入重新排列
video = video.view(-1, video.shape[-1])