def __init__(self):
     super().__init__()
     self.imagegradients = nn.ImageGradients()
Exemplo n.º 2
0
 def __init__(self):
     super(Net, self).__init__()
     self.image_gradients = nn.ImageGradients()