当前位置: 代码迷 >> 综合 >> pytorch RuntimeError: running_mean should contain 1048576 elements not 25088
  详细解决方案

pytorch RuntimeError: running_mean should contain 1048576 elements not 25088

热度:62   发布时间:2023-12-15 16:06:50.0

BN层的shape没有匹配好
只能修改输入输出的维度,仔细尝试

  相关解决方案