BN层的shape没有匹配好
只能修改输入输出的维度,仔细尝试
pytorch RuntimeError: running_mean should contain 1048576 elements not 25088
最新推荐文章于 2023-05-10 10:58:41 发布
BN层的shape没有匹配好
只能修改输入输出的维度,仔细尝试