2016-12-05 13 views

Trả lời

9

Bạn có thể làm điều đó miễn là mô hình của bạn không có phân bổ thiết bị rõ ràng. IE, nếu mô hình của bạn có các khối như with tf.device('gpu:0'), nó sẽ khiếu nại khi bạn chạy trên mô hình không có GPU.

Trong trường hợp này bạn phải chắc chắn mô hình nhập khẩu của bạn không có nhiệm vụ thiết bị rõ ràng, ví dụ, nhưng sử dụng clear_devices luận trong import_meta_graph

+0

Chỉ cần chắc chắn, nó có nghĩa là so với clear_devices = True tôi có thể sử dụng của tôi mô hình, ngay cả khi tôi đào tạo nó với các khối như với tf.device ('gpu: 0'), trên CPU của tôi? –

+0

có, xóa thiết bị và cho phép thuật toán vị trí tự động tf sử dụng các thiết bị có sẵn –

+0

cũng có thể lưu mô hình với thiết bị đã xóa, tới chuỗi tuần tự SavedModel: xem, ví dụ 'add_meta_graph()' trong [SavedModelBuilder] (https: //www.tensorflow.org/api_docs/python/tf/saved_model/builder/SavedModelBuilder) lớp. – Mark

Các vấn đề liên quan