RuntimeError: running_mean should contain 250 elements not 240

 self.layer2 = nn.BatchNorm1d(250)

将参数240 改为250 既可以

发布了234 篇原创文章 · 获赞 61 · 访问量 12万+

猜你喜欢

转载自blog.csdn.net/weixin_42528089/article/details/104855690