From 42c076edaf071e7f5a5f9154e1f3c1302c038aba Mon Sep 17 00:00:00 2001 From: "Zhengyi \"Zen\" Luo" Date: Sun, 17 Nov 2024 20:49:42 -0800 Subject: [PATCH] fixing get_mean_std_with_masks (#313) --- 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)