backup wip

2024-04-05 17:38:29 +01:00
parent 9c28ac8aa4
commit 1e71196fe3
7 changed files with 306 additions and 298 deletions
--- a/lerobot/common/policies/diffusion/policy.py
+++ b/lerobot/common/policies/diffusion/policy.py
@@ -152,7 +152,6 @@ class DiffusionPolicy(nn.Module):
        self.diffusion.train()

        data_s = time.time() - start_time
-
        loss = self.diffusion.compute_loss(batch)
        loss.backward()