Exemplo n.º 1
0
 def __init__(self):
     SparseModule.__init__(self)
     self.module = nn.Tanh()
     self.output = SparseConvNetTensor()
     self.output.features = self.module.output
     self.gradInput = self.module.gradInput
Exemplo n.º 2
0
 def callback(module):
     if isinstance(module, nn.ReLU):
         return nn.Tanh()
     return module
Exemplo n.º 3
0
 def test_tanh(self):
     self._test_single_layer(nn.Tanh())