From ed7bd0e0b56d09f9d932241ced7e65f487903999 Mon Sep 17 00:00:00 2001 From: zhengyiluo Date: Sun, 17 Nov 2024 02:44:55 -0500 Subject: [PATCH] fixing get_mean_std_with_masks --- rl_games/algos_torch/running_mean_std.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/rl_games/algos_torch/running_mean_std.py b/rl_games/algos_torch/running_mean_std.py index 9bedc544..b6996fd1 100644 --- a/rl_games/algos_torch/running_mean_std.py +++ b/rl_games/algos_torch/running_mean_std.py @@ -45,7 +45,7 @@ def _update_mean_var_count_from_moments(self, mean, var, count, batch_mean, batc def forward(self, input, denorm=False, mask=None): if self.training: if mask is not None: - mean, var = torch_ext.get_mean_std_with_masks(input, mask) + mean, var = torch_ext.get_mean_var_with_masks(input, mask) else: mean = input.mean(self.axis) # along channel axis var = input.var(self.axis)