Một máy chủ TensorFlow trong quá trình, để sử dụng trong đào tạo phân tán.
Một cá thể Server
đóng gói một tập hợp các thiết bị và một mục tiêu Session
có thể tham gia vào quá trình đào tạo phân tán. Máy chủ thuộc về một cụm (được chỉ định bởi ClusterSpec
) và tương ứng với một nhiệm vụ cụ thể trong một công việc được đặt tên. Máy chủ có thể giao tiếp với bất kỳ máy chủ nào khác trong cùng một cụm. Máy chủ sẽ không phục vụ bất kỳ yêu cầu nào cho đến khi start()
được gọi. Máy chủ sẽ ngừng cung cấp các yêu cầu khi stop()
hoặc close()
được gọi. Lưu ý rằng phương thức close()
dừng máy chủ nếu nó đang chạy.
CẢNH BÁO: Server
sở hữu tài nguyên phải được giải phóng rõ ràng bằng cách gọi hàm close()
.
Các phiên bản của Server
là an toàn theo luồng.
Sử dụng ví dụ:
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();
}
Người xây dựng công cộng
Máy chủ (byte [] serverDef) Xây dựng một phiên bản mới của máy chủ. |
Phương thức công khai
khoảng trống đồng bộ | đóng () Phá hủy một máy chủ TensorFlow đang trong quá trình, giải phóng bộ nhớ. |
vô hiệu | tham gia () Chặn cho đến khi máy chủ được dừng thành công. |
khoảng trống đồng bộ | bắt đầu () Khởi động máy chủ TensorFlow trong quá trình. |
khoảng trống đồng bộ | dừng lại () Dừng máy chủ TensorFlow đang trong quá trình. |
Phương thức kế thừa
Người xây dựng công cộng
Máy chủ công cộng (byte [] serverDef)
Xây dựng một phiên bản mới của máy chủ.
Thông số
serverDef | Định nghĩa máy chủ được chỉ định làm bộ đệm giao thức ServerDef được tuần tự hóa. |
---|
Phương thức công khai
công khai đồng bộ hóa void close ()
Phá hủy một máy chủ TensorFlow đang trong quá trình, giải phóng bộ nhớ.
Ném
Bị gián đoạn |
---|
public void join ()
Chặn cho đến khi máy chủ được dừng thành công.
công khai đồng bộ hóa void start ()
Khởi động máy chủ TensorFlow trong quá trình.
công khai đồng bộ hóa void stop ()
Dừng máy chủ TensorFlow đang trong quá trình.