소스 검색

print minimalistic remote server info

Tim Dettmers 3 년 전
부모
커밋
dd4e792937
1개의 변경된 파일3개의 추가작업 그리고 1개의 파일을 삭제
  1. 3 1
      src/client/remote_sequential.py

+ 3 - 1
src/client/remote_sequential.py

@@ -85,7 +85,6 @@ class RemoteSequential(nn.Module):
                 self.config,
                 self.dht,
                 prefix=self.prefix,
-                max_retries=self.max_retries,
                 p2p=self.p2p,
                 sequence_manager=self.sequence_manager[ix],
             )
@@ -100,3 +99,6 @@ class RemoteSequential(nn.Module):
     def inference_session(self) -> RemoteSequentialInferenceSession:
         self.sequence_manager.update_()
         return RemoteSequentialInferenceSession(self.sequence_manager, self.p2p)
+
+    def extra_repr(self) -> str:
+        return f"{self.sequence_manager.block_uids[0]}..{self.sequence_manager.block_uids[-1]}"