squeeze

kscalelabs · Dec 30, 2024 · 31dd379 · 31dd379
1 parent 8f120bb
commit 31dd379
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sim/model_export.py b/sim/model_export.py
@@ -186,8 +186,8 @@ def forward(
                 q,
                 dq,
                 prev_actions,
-                imu_ang_vel * self.ang_vel_scale,
-                imu_euler_xyz * self.quat_scale,
+                imu_ang_vel.squeeze(0) * self.ang_vel_scale,
+                imu_euler_xyz.squeeze(0) * self.quat_scale,
             ),
             dim=0,
         )