Pytorch里nn.BacthNorm1d(num_features)表示添加批标准化。其中num_features不是指的线性层的输出维度!!而是指输入数据的channel维度,一般DNN里肯定为1吧,DNN不会有通道数为3的输入吧,所以:nn.BatchNorm1d(num_features=1)

https://pytorch.org/docs/stable/nn.html#torch.nn.BatchNorm1d
在这里插入图片描述

可行的代码如下

self.model=nn.Sequential()
self.model.add_module('fc0',torch.nn.Linear(input_dimension,dim_hidden,bias=True))
self.model.add_module('bn0',nn.BatchNorm1d(num_features=1))
self.model.add_module('act0',activation(act_name))
       
for i in range(1,hidden_layers):
      self.model.add_module(f'fc{i}',nn.Linear(dim_hidden,dim_hidden,bias=True)) 
      self.model.add_module(f'bn{i}',nn.BatchNorm1d(num_features=1))    
      self.model.add_module(f'act{i}',activation(act_name))
      self.model.add_module(f'fc{hidden_layers}',nn.Linear(dim_hidden,output_dimension,bias=True))

注:我的输入数据维度为10,利用Dataloader分batch 取数据,输入数据的处理

features=torch.Tensor(features).view(1,10)
Logo

开放原子开发者工作坊旨在鼓励更多人参与开源活动,与志同道合的开发者们相互交流开发经验、分享开发心得、获取前沿技术趋势。工作坊有多种形式的开发者活动,如meetup、训练营等,主打技术交流,干货满满,真诚地邀请各位开发者共同参与!

更多推荐