コード例 #1
0
 def __init__(self):
     super().__init__(derivatives=Conv3DDerivatives(),
                      params=["bias", "weight"])
コード例 #2
0
ファイル: convnd.py プロジェクト: f-dangel/backpack
 def __init__(self):
     """Initialization."""
     super().__init__(["bias", "weight"], derivatives=Conv3DDerivatives())
コード例 #3
0
 def __init__(self):
     """Pass derivatives for ``torch.nn.Conv3d`` module."""
     super().__init__(Conv3DDerivatives(), params=["bias", "weight"])
コード例 #4
0
ファイル: conv3d.py プロジェクト: f-dangel/backpack
 def __init__(self):
     """Store parameter names and derivatives object."""
     super().__init__(derivatives=Conv3DDerivatives(),
                      params=["bias", "weight"])