Skip to content

Commit 3764890

Browse files
committed
Tune end_s for chunk size 100
1 parent 2a00839 commit 3764890

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/lerobot/policies/smolvla/modeling_smolvla.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -971,7 +971,7 @@ def rtc_denoise(self, noise, bsize, dt, prefix_pad_masks, past_key_values, devic
971971
grad_time = 0
972972

973973
# s_end = s
974-
s_end = 150
974+
s_end = 70
975975
# Prepare the guidance mask
976976
W = make_soft_mask(d, s_end, H, device)
977977
W_row = W[None, :, None] # broadcast to (B,H,M)

0 commit comments

Comments
 (0)