self.layer2 = nn.BatchNorm1d(250)

将参数240 改为250 既可以

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐