🔧 Fix localscaler bug (#70)

TakuyaShintate · Sep 26, 2021 · 8be7469 · 8be7469
1 parent 860f807
commit 8be7469
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tsts/models/localscalers/noop.py b/tsts/models/localscalers/noop.py
@@ -25,4 +25,4 @@ def from_cfg(
         return local_scaler
 
     def forward(self, bias: Tensor) -> Tensor:
-        return torch.zeros_like(bias)[:, :, 0:1]
+        return bias.new_zeros((1,))