mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-14 21:41:59 +00:00
Fix traacking
This commit is contained in:
@@ -40,7 +40,7 @@ class RTCConfig:
|
||||
# Core RTC settings
|
||||
# Todo change to exp
|
||||
prefix_attention_schedule: RTCAttentionSchedule = RTCAttentionSchedule.LINEAR
|
||||
max_guidance_weight: float = 5.0
|
||||
max_guidance_weight: float = 10.0
|
||||
execution_horizon: int = 10
|
||||
|
||||
# Debug settings
|
||||
|
||||
@@ -260,7 +260,6 @@ class RTCProcessor:
|
||||
x1_t = x1_t.squeeze(0)
|
||||
err = err.squeeze(0)
|
||||
|
||||
# Record debug information (all params except x_t which is recorded externally)
|
||||
self.track(
|
||||
time=time,
|
||||
x1_t=x1_t,
|
||||
|
||||
@@ -798,12 +798,13 @@ class VLAFlowMatching(nn.Module):
|
||||
|
||||
# Euler step
|
||||
x_t += dt * v_t
|
||||
time += dt
|
||||
|
||||
# Record x_t and v_t after Euler step (other params are recorded in rtc_processor.denoise_step)
|
||||
if self.rtc_processor is not None and self.rtc_processor.is_debug_enabled():
|
||||
self.rtc_processor.track(time=time, x_t=x_t, v_t=v_t)
|
||||
|
||||
time += dt
|
||||
|
||||
return x_t
|
||||
|
||||
def denoise_step(
|
||||
|
||||
Reference in New Issue
Block a user