Fixed the bug in flow matching (#68)

* fix a bug in flow matching * fixed the bugs in flow matching * fixed format --------- Co-authored-by: Geoffrey19 <sympathischmann35@gmail.com>
2025-11-07 15:11:49 +08:00
parent fbe354c266
commit 7b57b7800b
2 changed files with 2 additions and 2 deletions
@@ -307,7 +307,7 @@ class ActionProcessor(nn.Module):
            torch.Tensor: Sampled timesteps of shape [batch_size]
        """
        sample = self.beta_dist.sample([batch_size]).to(device=device, dtype=dtype)
-        time = (1 - sample) / self.s
+        time = (1 - sample) * self.s
        return time

    def proprioception_proj(