소스 검색

Fix sequential_forward()

Aleksandr Borzunov 3 년 전
부모
커밋
756e27707f
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      src/client/sequential_autograd.py

+ 1 - 1
src/client/sequential_autograd.py

@@ -48,7 +48,7 @@ async def sequential_forward(
     outputs = inputs
 
     block_idx = start_index
-    while block_idx < len(sequence_manager):
+    while block_idx < end_index:
         for attempt_no in itertools.count():
             logger.debug(f"Forward: block {block_idx}, attempt {attempt_no}")
             try: