Внутрипроцессный сервер TensorFlow для использования в распределенном обучении.
Экземпляр Server
инкапсулирует набор устройств и цель Session
которые могут участвовать в распределенном обучении. Сервер принадлежит к кластеру (указанному ClusterSpec
) и соответствует определенной задаче в названном задании. Сервер может взаимодействовать с любым другим сервером в том же кластере. Сервер не будет обслуживать запросы, пока не будет вызвана функцияstart()
. Сервер перестанет обслуживать запросы после вызова stop()
илиclose()
. Имейте в виду, что методclose()
останавливает сервер, если он запущен.
ВНИМАНИЕ: Server
владеет ресурсами, которые должны быть явно освобождены с помощью вызоваclose()
.
Экземпляры 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. |
Унаследованные методы
Общественные конструкторы
Публичные методы
общедоступная синхронизированная пустота close ()
Уничтожить внутрипроцессный сервер TensorFlow, освободить память.
Броски
InterruptedException |
---|
public void join ()
Блокирует до тех пор, пока сервер не будет успешно остановлен.
общедоступный синхронизированный void start ()
Запускает внутрипроцессный сервер TensorFlow.
общедоступная синхронизированная пустая остановка ()
Останавливает внутрипроцессный сервер TensorFlow.