Tôi đang cố gắng để tạo ra một tập hợp của nhiều mô hình được đào tạo. Tất cả các mô hình có cùng một biểu đồ và chỉ khác nhau theo trọng số của nó. Tôi đang tạo biểu đồ mô hình bằng cách sử dụng tf.get_variable
. Tôi có một số điểm kiểm tra khác nhau (với các trọng số khác nhau) cho cùng một kiến trúc đồ thị và tôi muốn tạo một mô hình mẫu cho mỗi trạm kiểm soát.Làm thế nào để tạo ra toàn bộ trong tensorflow?
Làm cách nào tôi có thể tải nhiều điểm kiểm tra mà không ghi đè trọng số đã tải trước đó?
Khi tôi tạo biểu đồ của mình với tf.get_variable
, cách duy nhất tôi có thể tạo nhiều biểu đồ là bằng cách chuyển đối số reuse = True
. Bây giờ nếu tôi thử thay đổi tên của các biến đồ thị của tôi kèm theo phương thức xây dựng trong một phạm vi mới (để chúng trở thành không thể chia sẻ với các đồ thị được tạo khác) trước khi tải, thì điều này sẽ không hoạt động vì tên mới sẽ khác với trọng lượng và tôi sẽ không thể tải nó.
Tôi chưa thử, nhưng dưới đây là một số mã tham chiếu: https://github.com/eske/seq2seq/blob/master/translate/__main__.py#L190 Tóm lại, tác giả tạo ra nhiều phiên làm nhiều điểm kiểm tra và trong mỗi phiên, anh ta khôi phục điểm kiểm tra tương ứng. – tnq177
Một số mã tham chiếu có liên quan hơn: https://github.com/Hvass-Labs/TensorFlow-Tutorials/blob/master/05_Ensemble_Learning.ipynb – Deepank
@cesarsalgado: Tôi có cùng một vấn đề. Tôi đang sử dụng inception-v4 trong tf-slim. Bạn đã giải quyết nó như thế nào? – Jame