Um servidor TensorFlow em processo, para uso em treinamento distribuído.
Uma instância de Server
encapsula um conjunto de dispositivos e um destino de Session
que pode participar de treinamento distribuído. Um servidor pertence a um cluster (especificado por um ClusterSpec
) e corresponde a uma tarefa específica em um trabalho nomeado. O servidor pode se comunicar com qualquer outro servidor no mesmo cluster. O servidor não atenderá a nenhuma solicitação até que start()
seja invocado. O servidor irá parar de atender às solicitações assim que stop()
ou close()
for invocado. Esteja ciente de que o método close()
para o servidor se ele estiver em execução.
AVISO: Um Server
possui recursos que devem ser liberados explicitamente invocando close()
.
As instâncias de um Server
são thread-safe.
Usando o exemplo:
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();
}
Construtores Públicos
Servidor (byte [] serverDef) Constrói uma nova instância de servidor. |
Métodos Públicos
vazio sincronizado | fechar () Destrua um servidor TensorFlow em processo e libera memória. |
vazio | juntar () Bloqueia até que o servidor seja interrompido com sucesso. |
vazio sincronizado | start () Inicia um servidor TensorFlow em processo. |
vazio sincronizado | parar () Para um servidor TensorFlow em processo. |
Métodos herdados
Construtores Públicos
servidor público (byte [] serverDef)
Constrói uma nova instância de servidor.
Parâmetros
serverDef | Definição de servidor especificada como um buffer de protocolo ServerDef serializado. |
---|
Métodos Públicos
public synchronized void close ()
Destrua um servidor TensorFlow em processo e libera memória.
Lança
InterruptedException |
---|
public void join ()
Bloqueia até que o servidor seja parado com sucesso.
public synchronized void start ()
Inicia um servidor TensorFlow em processo.
public synchronized void stop ()
Para um servidor TensorFlow em processo.