amend

pytorch · Nov 22, 2023 · 4edd592 · 4edd592
1 parent 1a2e65e
commit 4edd592
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tensordict/base.py b/tensordict/base.py
@@ -3041,6 +3041,7 @@ def map(
         imap = pool.imap(fn, self_split, chunksize)
         while len(out) < len(self_split):
             out.append(imap.next(timeout=50.0))
+            print(len(out))
         # out = pool.map(fn, self_split, chunksize)
         out = torch.cat(list(imap), dim)
         return out