После обучения более сложной модели, я бы предпочел сохранить только некоторые ее элементы.Конечно, я могу сделать это с переменными тензорного потока.Но два конкретных элемента графа являются для меня проблемой: LSTMCell (LSTMBlockCell, если это имеет какое-либо значение) и плотный слой (tf.layers.dense ()).
some_lstm_cell = tf.contrib.rnn.LSTMBlockCell(hidden_units, name='my_cell')
dense_layer = tf.layers.dense(input, num_units, name='my_dense_layer')
Я специально не хочусохранить всю модель, а лучше только сохранить, а затем восстановить эти конкретные элементы.по какой-то причине.
Передача их в качестве элементов списка в аргументе var_list функции tf.train.Saver () приводит к возникновению ошибки (что имеет смысл, ни одна из них не является переменной).(Я могу добавить сообщения об ошибках, если кому-то интересно).
Как я могу продолжить, сохранить и восстановить эти конкретные элементы?Есть две проблемы: - я бы предпочел использовать tf.train.Saver (), но если другие способы проще, я хотел бы знать их - я хочу иметь ссылку на оба элемента после их восстановления
Казалось бы, кто-то должен был задавать этот вопрос раньше, но сейчас я ищу около двух часов и до сих пор ничего не нашел, поэтому я все равно решил опубликовать этот вопрос.