fix(roland): fix typo in model/utils.py (#290)

opendilab · Nov 5, 2024 · edee973 · edee973
1 parent 761bfa9
commit edee973
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lzero/model/utils.py b/lzero/model/utils.py
@@ -117,7 +117,7 @@ def renormalize(inputs: torch.Tensor, first_dim: int = 1) -> torch.Tensor:
     min_val = torch.min(flat_input, first_dim, keepdim=True).values
     flat_input = (flat_input - min_val) / (max_val - min_val)
 
-    return flat_input.view(*input.shape)
+    return flat_input.view(*inputs.shape)
 
 
 def get_dynamic_mean(model: nn.Module) -> float: