Sử dụng bộ sưu tập để sắp xếp ngăn nắp các trang
Lưu và phân loại nội dung dựa trên lựa chọn ưu tiên của bạn.
Giới thiệu
TensorFlow Hub lưu trữ các mô hình cho nhiều nhiệm vụ khác nhau. Các mô hình cho cùng một nhiệm vụ được khuyến khích triển khai một API chung để người tiêu dùng mô hình có thể dễ dàng trao đổi chúng mà không cần sửa đổi mã sử dụng chúng, ngay cả khi chúng đến từ các nhà xuất bản khác nhau.
Mục tiêu là làm cho việc trao đổi các mô hình khác nhau cho cùng một nhiệm vụ trở nên đơn giản như chuyển đổi một siêu tham số có giá trị chuỗi. Cùng với đó, người tiêu dùng mô hình có thể dễ dàng tìm thấy mô hình tốt nhất cho vấn đề của họ.
API SavingModel có thể tái sử dụng xác định các quy ước chung về cách tải SavingModel trở lại chương trình Python và tái sử dụng nó như một phần của mô hình TensorFlow lớn hơn.
Cách sử dụng cơ bản:
obj = hub.load("path/to/model") # That's tf.saved_model.load() after download.
outputs = obj(inputs, training=False) # Invokes the tf.function obj.__call__.
Đối với người dùng Keras, lớp hub.KerasLayer dựa vào API này để bao bọc SavingModel có thể tái sử dụng dưới dạng Lớp Keras (bảo vệ người dùng Keras khỏi các chi tiết của nó), với đầu vào và đầu ra theo các API dành riêng cho nhiệm vụ được liệt kê bên dưới.