瀏覽代碼

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]}"