분산 교육에 사용하기 위한 in-process 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();
}
공공 생성자
서버 (바이트[] serverDef) 서버의 새 인스턴스를 구성합니다. |
공개 방법
동기화된 무효 | 닫다 () 진행 중인 TensorFlow 서버를 삭제하고 메모리를 확보합니다. |
무효의 | 가입하다 () 서버가 성공적으로 중지될 때까지 차단합니다. |
동기화된 무효 | 시작 () 진행 중인 TensorFlow 서버를 시작합니다. |
동기화된 무효 | 멈추다 () 진행 중인 TensorFlow 서버를 중지합니다. |
상속된 메서드
공공 생성자
공개 방법
공개 동기화 무효 닫기 ()
진행 중인 TensorFlow 서버를 파괴하고 메모리를 확보합니다.
던지기
중단된 예외 |
---|
공개 무효 가입 ()
서버가 성공적으로 중지될 때까지 차단합니다.
공개 동기화 무효 시작 ()
진행 중인 TensorFlow 서버를 시작합니다.
공개 동기화 무효 중지 ()
진행 중인 TensorFlow 서버를 중지합니다.