Преглед на файлове

mention reference issue

Your Name преди 1 година
родител
ревизия
9e29140bb0
променени са 1 файла, в които са добавени 2 реда и са изтрити 2 реда
  1. 2 2
      src/petals/client/remote_forward_backward.py

+ 2 - 2
src/petals/client/remote_forward_backward.py

@@ -144,8 +144,8 @@ async def run_remote_backward(
             for tensor, compression in zip(flat_tensors, codecs)
         )
     )
-    for tensor, serialized_tensor in zip(flat_tensors, serialized_tensors):
-        serialized_tensor.requires_grad = tensor.requires_grad
+    for tensor, serialized in zip(flat_tensors, serialized_tensors):
+        serialized.requires_grad = tensor.requires_grad  # see https://github.com/learning-at-home/hivemind/pull/591
 
     size = sum(t.element_size() * t.nelement() for t in flat_tensors)
     backward_fn = _backward_stream if size > MAX_UNARY_PAYLOAD_SIZE // 2 else _backward_unary