huangyanpei/VLA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix normalization (#112)
Pre-commit / pre-commit (push) Canceled after 0s
Details

Browse Source
This commit is contained in:
Zihan Gao
2026-07-21 15:15:48 +08:00
committed by GitHub
parent 70dcf78816
commit 6764e8f12f
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
wall_x/model/core/action/normalizer.py
+3 -3
View File
@@ -285,9 +285,9 @@ class Normalizer(nn.Module):
new_xs = [] new_xs = []
dataset_names = [name for name in dataset_names if name != "x2_multimodal"] dataset_names = [name for name in dataset_names if name != "x2_multimodal"]
for x, dataset_name in zip(xs, dataset_names): for x, dataset_name in zip(xs, dataset_names):
# if dataset_name == "ex_normal": delta = self.delta[dataset_name]
# dataset_name = "x2_normal" delta = torch.where(delta == 0, torch.ones_like(delta), delta)
x = (x - self.min[dataset_name]) / (self.delta[dataset_name]) x = (x - self.min[dataset_name]) / delta
x = x * 2 - 1 x = x * 2 - 1
x = torch.clamp(x, -1, 1) x = torch.clamp(x, -1, 1)
new_xs.append(x) new_xs.append(x)