BatchNorm1d (10, momentum=0.5) setattr(self, 'bn%i' % i, bn) # IMPORTANT set layer to the Module self.bns.append(bn) for epoch in ...
確定! 回上一頁