分散トレーニングで使用するためのインプロセスTensorFlowサーバー。
Server
インスタンスは、デバイスのセットと、カプセル化Session
分散訓練に参加することができ、目標を。サーバーはクラスター( ClusterSpec
指定)に属し、名前付きジョブの特定のタスクに対応します。サーバーは、同じクラスター内の他のサーバーと通信できます。サーバーは、 start()
が呼び出されるまで要求を処理しません。 stop()
またはclose()
が呼び出されると、サーバーは要求の処理を停止します。 close()
メソッドが実行中の場合、サーバーを停止することclose()
注意してください。
警告: Server
は、 close()
を呼び出すことによって明示的に解放する必要があるリソースを所有していServer
。
Server
インスタンスはスレッドセーフです。
使用例:
import org.tensorflow.Server;
import org.tensorflow.distruntime.ClusterDef;
import org.tensorflow.distruntime.JobDef;
import org.tensorflow.distruntime.ServerDef;
ClusterDef clusterDef = ClusterDef.newBuilder()
.addJob(JobDef.newBuilder()
.setName("worker")
.putTasks(0, "localhost:4321")
.build()
).build();
ServerDef serverDef = ServerDef.newBuilder()
.setCluster(clusterDef)
.setJobName("worker")
.setTaskIndex(0)
.setProtocol("grpc")
.build();
try (Server srv = new Server(serverDef.toByteArray())) {
srv.start();
srv.join();
}
パブリックコンストラクター
サーバー(byte [] serverDef) サーバーの新しいインスタンスを構築します。 |
パブリックメソッド
同期ボイド | 近いです() インプロセスのTensorFlowサーバーを破棄し、メモリを解放します。 |
ボイド | 参加() サーバーが正常に停止するまでブロックします。 |
同期ボイド | 開始() インプロセスのTensorFlowサーバーを起動します。 |
同期ボイド | ストップ() 処理中のTensorFlowサーバーを停止します。 |
継承されたメソッド
パブリックコンストラクター
パブリックサーバー(byte [] serverDef)
サーバーの新しいインスタンスを構築します。
パラメーター
serverDef | シリアル化されたServerDefプロトコルバッファとして指定されたサーバー定義。 |
---|
パブリックメソッド
パブリック同期ボイドクローズ()
インプロセスのTensorFlowサーバーを破棄し、メモリを解放します。
投げる
InterruptedException |
---|
public void join ()
サーバーが正常に停止するまでブロックします。
パブリック同期ボイドスタート()
インプロセスのTensorFlowサーバーを起動します。
パブリック同期ボイドストップ()
処理中のTensorFlowサーバーを停止します。