Method __init__

numpy_ml/tests/nn_torch_models.py:83–85

(self, lambda_=10)

class TorchWGANGPLoss(nn.Module):
    def __init__(self, lambda_=10):
        self.lambda_ = torchify([lambda_])
        super(TorchWGANGPLoss, self).__init__()

    def forward(self, Y_real, Y_fake, gradInterp):
        GY_fake = Y_fake.copy()