Máy chủ TensorFlow đang trong quá trình sử dụng để đào tạo phân tán.
Phiên bản Server
đóng gói một tập hợp thiết bị và mục tiêu Session
có thể tham gia đào tạo phân tán. Một máy chủ thuộc về một cụm (được chỉ định bởi ClusterSpec
) và tương ứng với một tác vụ cụ thể trong 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 sau khi gọi lệnh stop()
hoặc close()
. Xin lưu ý rằng phương thức close()
sẽ dừng máy chủ nếu nó đang chạy.
CẢNH BÁO: Server
sở hữu các tài nguyên phải được giải phóng rõ ràng bằng cách gọi close()
.
Các phiên bản của Server
đều 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();
}
Nhà 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 pháp công cộng
khoảng trống đồng bộ | đóng () Phá hủy máy chủ TensorFlow đang xử lý, giải phóng bộ nhớ. |
trống rỗng | 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 đang xử lý. |
khoảng trống đồng bộ | dừng lại () Dừng máy chủ TensorFlow đang xử lý. |
Phương pháp kế thừa
Nhà 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ố
máy chủDef | Đị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 pháp công cộng
đóng đồng bộ công khai void ()
Phá hủy máy chủ TensorFlow đang xử lý, giải phóng bộ nhớ.
Ném
Ngoại lệ bị gián đoạn |
---|
tham gia void công khai ()
Chặn cho đến khi máy chủ được dừng thành công.
bắt đầu khoảng trống được đồng bộ hóa công khai ()
Khởi động máy chủ TensorFlow đang xử lý.
dừng đồng bộ công khai ()
Dừng máy chủ TensorFlow đang xử lý.