From f6585d4059640bba3b79849b09d6c9c0b4e04914 Mon Sep 17 00:00:00 2001 From: irisliucy Date: Mon, 8 Mar 2021 13:13:42 -0800 Subject: [PATCH] Disable validate_args --- src/garage/torch/distributions/tanh_normal.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/garage/torch/distributions/tanh_normal.py b/src/garage/torch/distributions/tanh_normal.py index 6d1ca10d34..51a835031c 100644 --- a/src/garage/torch/distributions/tanh_normal.py +++ b/src/garage/torch/distributions/tanh_normal.py @@ -19,7 +19,7 @@ class TanhNormal(torch.distributions.Distribution): """ # noqa: 501 def __init__(self, loc, scale): - self._normal = Independent(Normal(loc, scale), 1) + self._normal = Independent(Normal(loc, scale), 1, validate_args=False) super().__init__() def log_prob(self, value, pre_tanh_value=None, epsilon=1e-6):