Server

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();
 }