Method __init__

numpy_ml/tests/nn_torch_models.py:52–53

(self)

class TorchVAELoss(nn.Module):
    def __init__(self):
        super(TorchVAELoss, self).__init__()

    def extract_grads(self, X, X_recon, t_mean, t_log_var):
        eps = np.finfo(float).eps