Phần này chứa các bộ sưu tập bổ sung của các trang tham chiếu API cho các dự án và gói tách biệt với gói tensorflow , nhưng không có các trang phụ dành riêng.
Kho lưu trữ Mô hình TensorFlow
Kho lưu trữ Mô hình TensorFlow cung cấp triển khai các mô hình hiện đại (SOTA).
Thư mục chính thức / dự án chứa một bộ sưu tập các mô hình SOTA sử dụng API cấp cao của TensorFlow. Chúng nhằm mục đích được duy trì tốt, được kiểm tra và luôn cập nhật với API TensorFlow mới nhất.
Mã thư viện được sử dụng để xây dựng và đào tạo các mô hình này có sẵn dưới dạng gói pip. Bạn có thể cài đặt nó bằng cách sử dụng:
$ pip install tensorflow-models-official # For the latest release
$ #or
$ pip install tf-models-nightly # For the nightly build
Để cài đặt gói từ nguồn, hãy tham khảo các hướng dẫn này .
Gói pip tensorflow-models-official chứa hai mô-đun cấp cao nhất: tensorflow_models và orbit . Bạn có thể nhập chúng bằng:
import tensorflow_models as tfm
import orbit
Mô hình Tensorflow
Mô-đun tensorflow_models xử lý xây dựng mô hình và đào tạo cấu hình. Chức năng dành riêng cho ứng dụng có sẵn trong tfm.vision và tfm.nlp .
Quỹ đạo
Mô-đun orbit xác định một thư viện linh hoạt và nhẹ để viết mã vòng lặp đào tạo tùy chỉnh trong TensorFlow. Orbit linh hoạt về loại mô hình mà nó hoạt động. Bạn có thể sử dụng Orbit để đào tạo Keras Models (thay thế cho Keras ' Model.fit ), nhưng bạn hoàn toàn không phải sử dụng Keras. Orbit tích hợp hoàn toàn với tf.distribute và hỗ trợ chạy trên các loại thiết bị khác nhau (CPU, GPU và TPU).