From 47de0896908a5a68b6326c32d188ca31c9fd3de1 Mon Sep 17 00:00:00 2001 From: Tim Joseph Date: Thu, 11 Jul 2024 13:35:53 +0200 Subject: [PATCH] Removed global set_default_validate_args Having it true deviated from the default PyTorch behavior. --- tensordict/nn/distributions/continuous.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/tensordict/nn/distributions/continuous.py b/tensordict/nn/distributions/continuous.py index a511205c4..b6135833d 100644 --- a/tensordict/nn/distributions/continuous.py +++ b/tensordict/nn/distributions/continuous.py @@ -21,9 +21,6 @@ "Delta", ] -# speeds up distribution construction -D.Distribution.set_default_validate_args(False) - class NormalParamWrapper(nn.Module): """A wrapper for normal distribution parameters.