Python BatchNorm1dDerivatives示例

示例#1

文件： batchnorm1d.py 项目： toiaydcdyywlhzvlob/backpack

 def __init__(self):
     super().__init__(derivatives=BatchNorm1dDerivatives(),
                      params=["bias", "weight"])

示例#2

 def __init__(self, damping=1.0):
     self.damping = damping
     super().__init__(derivatives=BatchNorm1dDerivatives(), params=["bias", "weight"])

示例#3

 def __init__(self, silent):
     self.silent = silent
     super().__init__(derivatives=BatchNorm1dDerivatives(),
                      params=["bias", "weight"])