PyTorch -无法将batchnorm1d与线性一起使用
PyTorch -无法将batchnorm1d与线性一起使用
提问于 2019-06-01 01:05:33
在一维张量上使用PyTorch的BatchNorm1D会给出错误：
RuntimeError: running_mean应包含%1个不是2304的元素
有什么建议可能是错误的吗？
我的代码：
self.net_common = nn.Sequential (
nn.Linear(64*64, 48*48),
nn.BatchNorm1d(48*48),
nn.Tanh(),
nn.Dropout(p=0.25),
nn.Linear(48*48, 32*32),
nn.BatchNorm1d(32*32),
nn.Tanh(),
)页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/56399151
复制相关文章
相似问题