Um servidor TensorFlow em processo, para uso em treinamento distribuído.
Uma instância Server
encapsula um conjunto de dispositivos e um alvo Session
que pode participar de treinamento distribuído. Um servidor pertence a um cluster (especificado por 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á nenhuma solicitação até que start()
seja invocado. O servidor irá parar de atender solicitações assim que stop()
ou close()
for invocado. Esteja ciente de que o método close()
interrompe o servidor se ele estiver em execução.
AVISO: Um Server
possui recursos que devem ser explicitamente liberados 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 do servidor. |
Métodos Públicos
vazio sincronizado | fechar () Destrua um servidor TensorFlow em processo e libere memória. |
vazio | juntar () Bloqueia até que o servidor seja parado com sucesso. |
vazio sincronizado | começar () Inicia um servidor TensorFlow em processo. |
vazio sincronizado | parar () Interrompe um servidor TensorFlow em processo. |
Métodos herdados
Construtores Públicos
Servidor público (byte[] serverDef)
Constrói uma nova instância do servidor.
Parâmetros
servidorDef | Definição de servidor especificada como um buffer de protocolo ServerDef serializado. |
---|
Métodos Públicos
fechamento de vazio sincronizado público ()
Destrua um servidor TensorFlow em processo e libere memória.
Lança
Exceção Interrompida |
---|
junção nula pública ()
Bloqueia até que o servidor seja parado com sucesso.
início vazio sincronizado público ()
Inicia um servidor TensorFlow em processo.
parada nula sincronizada pública ()
Interrompe um servidor TensorFlow em processo.