SAC works

2025-01-14 11:34:52 +01:00
parent 86df8a433d
commit c1d4bf4b63
2 changed files with 6 additions and 0 deletions
--- a/lerobot/scripts/train_sac.py
+++ b/lerobot/scripts/train_sac.py
@@ -265,6 +265,9 @@ class ReplayBuffer:
        batch_dones = torch.tensor([t["done"] for t in list_of_transitions], dtype=torch.float32).to(
            self.device
        )
+        batch_dones = torch.tensor([t["done"] for t in list_of_transitions], dtype=torch.float32).to(
+            self.device
+        )

        # Return a BatchTransition typed dict
        return BatchTransition(