SavedModelBundle

lớp công khai SavingModelBundle

SavingModelBundle đại diện cho một mô hình được tải từ bộ lưu trữ.

Mô hình bao gồm mô tả tính toán ( Graph ), Session có các tensor (ví dụ: tham số hoặc biến trong biểu đồ) được khởi tạo thành các giá trị được lưu trong bộ lưu trữ và mô tả mô hình dưới dạng bộ đệm giao thức MetaGraphDef .

Các lớp lồng nhau

lớp học Đã lưuModelBundle.Exporter Các tùy chọn để xuất SavingModel.
lớp học Đã lưuModelBundle.Loader Các tùy chọn để tải SavingModel.

Hằng số

Sợi dây DEFAULT_TAG

Phương pháp công cộng

Bản đồ<Chuỗi, Tensor >
gọi (Bản đồ<String, Tensor > đối số)
Gọi hàm mặc định trực tiếp từ mô hình này.
trống rỗng
đóng ()
Phát hành các tài nguyên ( Graph và Session ) được liên kết với gói mô hình đã lưu.
tĩnh SavingModelBundle.Exporter
nhà xuất khẩu (Chuỗi xuấtDir)
Xuất mô hình đã lưu.
bê tôngChức năng
hàm (Chuỗi signatureKey)
Trả về ConcreteFunction tương ứng với chữ ký hàm.
đồ thị
đồ thị ()
Trả về biểu đồ mô tả tính toán được thực hiện bởi mô hình.
Đã lưu tĩnhModelBundle
tải (Thẻ xuất chuỗiDir, Chuỗi...)
Tải mô hình đã lưu từ thư mục xuất.
tĩnh SavingModelBundle.Loader
trình tải (Xuất chuỗiDir)
Tải một mô hình đã lưu.
MetaGraphDef
metaGraphDef ()
Trả về bộ đệm giao thức MetaGraphDef được liên kết với mô hình đã lưu.
Phiên họp
phiên họp ()
Trả về Session để thực hiện tính toán bằng mô hình.
Danh sách< Chữ ký >
chữ ký ()
Trả về chữ ký của tất cả các chức năng có sẵn trong mô hình đã lưu này.

Phương pháp kế thừa

Hằng số

Chuỗi cuối cùng tĩnh công khai DEFAULT_TAG

Giá trị không đổi: "phục vụ"

Phương pháp công cộng

public Map<String, Tensor > call (Bản đồ<String, Tensor > đối số)

Gọi hàm mặc định trực tiếp từ mô hình này.

Việc lựa chọn hàm mặc định được thực hiện dựa trên điều kiện đầu tiên đúng sau đây:

  • Hàm này là chữ ký duy nhất có sẵn được đính kèm vào biểu đồ chính của mô hình đã lưu này
  • Hàm này được ánh xạ tới tên chữ ký mặc định là "phục vụ_default"

Người gọi có trách nhiệm đóng tất cả các Tensors được trả về.

Thông số
lý lẽ danh sách các tensor đầu vào, được ánh xạ theo tên chữ ký của chúng
Trả lại
  • danh sách các tensor đầu ra, được ánh xạ theo tên chữ ký
Ném
Đối số bất hợp phápNgoại lệ nếu không có chức năng nào có thể được chọn theo mặc định

đóng khoảng trống công khai ()

Phát hành các tài nguyên ( Graph và Session ) được liên kết với gói mô hình đã lưu.

nhà xuất khẩu tĩnh công khai SavingModelBundle.Exporter (Chuỗi xuấtDir)

Xuất mô hình đã lưu.

Trả về đối tượng Exporter để thiết lập các tùy chọn cấu hình trước khi thực sự lưu mô hình.

Thông số
xuấtDir đường dẫn thư mục chứa mô hình đã lưu.

hàm ConcreteFunction công khai (String signatureKey)

Trả về ConcreteFunction tương ứng với chữ ký hàm.

ConcreteFunction myFunction = savedModelBundle.function("mySignatureKey");
 Map<String, Tensor> outputTensorMap = myFunction.call(session, inputTensorMap);

Thông số
chữ kýKey tên của SignatureDef trong mô hình đã lưu.
Trả lại
  • đối tượng có thể được sử dụng để thực hiện cuộc gọi đến một hàm
Ném
Đối số bất hợp phápNgoại lệ nếu không tìm thấy signatureKey trong mô hình đã lưu này.

biểu đồ đồ thị công khai ()

Trả về biểu đồ mô tả tính toán được thực hiện bởi mô hình.

Tải tĩnh công khai SavingModelBundle (thẻ String importDir, String...)

Tải mô hình đã lưu từ thư mục xuất. Mô hình đang được tải phải được tạo bằng API mô hình đã lưu .

Phương pháp này là viết tắt của:

SavedModelBundle.loader().withTags(tags).load();

Thông số
xuấtDir đường dẫn thư mục chứa mô hình đã lưu.
thẻ các thẻ xác định metagraphdef cụ thể để tải.
Trả lại
  • một gói chứa biểu đồ và phiên liên quan.

Trình tải tĩnh công khai SavingModelBundle.Loader (Chuỗi xuấtDir)

Tải một mô hình đã lưu.

Trả về một đối tượng Loader có thể đặt các tùy chọn cấu hình trước khi thực sự tải mô hình,

Thông số
xuấtDir đường dẫn thư mục chứa mô hình đã lưu.

metaGraphDef công khai metaGraphDef ()

Trả về bộ đệm giao thức MetaGraphDef được liên kết với mô hình đã lưu.

phiên phiên công khai ()

Trả về Session để thực hiện tính toán bằng mô hình.

Trả lại
  • phiên khởi tạo

Danh sách công khai< Chữ ký > chữ ký ()

Trả về chữ ký của tất cả các chức năng có sẵn trong mô hình đã lưu này.