From b42c1c68fd5cf328ed8d51dc12842d8f6605ef07 Mon Sep 17 00:00:00 2001
From: gouhanke <12219217+gouhanke@user.noreply.gitee.com>
Date: Wed, 11 Feb 2026 17:13:55 +0800
Subject: [PATCH] =?UTF-8?q?debug:=20=E5=B0=86=E5=BD=92=E4=B8=80=E5=8C=96?=
 =?UTF-8?q?=E6=94=BE=E5=9C=A8GPU=E4=B8=8A?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 roboimi/vla/data/simpe_robot_dataset.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/roboimi/vla/data/simpe_robot_dataset.py b/roboimi/vla/data/simpe_robot_dataset.py
index 4858e9d..ca690f4 100644
--- a/roboimi/vla/data/simpe_robot_dataset.py
+++ b/roboimi/vla/data/simpe_robot_dataset.py
@@ -86,9 +86,8 @@ class SimpleRobotDataset(Dataset):
                 h5_path = f'observations/images/{cam_name}'
                 if h5_path in f:
                     img = f[h5_path][meta["frame_idx"]]
-                    img = torch.from_numpy(img).float()
-                    # 归一化到 [0, 1] 范围（与推理时保持一致）
-                    img = img / 255.0
+                    img = torch.from_numpy(img)
+                    # 保持 uint8 格式以节省传输带宽，归一化移至 GPU (在 train_vla.py 中处理)
                     frame[f"observation.{cam_name}"] = img.permute(2, 0, 1)  # HWC -> CHW
 
         return frame