Как загрузить веса обученной модели, которые были сохранены с помощью tf.keras.models.save_model?

Question

Я обучил нейронную сеть без каких-либо контрольных точек, и в конце я написал tf.keras.models.save_model(model, dirpath) для сохранения всей модели, в результате чего были созданы следующие файлы:

savedmodel.pb 
assets/
variables/variables.index
variables/variables.data-00000-of-00001

Я попытался загрузить модель, используя new_model = tf.keras.models.load_model(dirpath), но он дал ValueError, потому что я использую пользовательскую модель (кажется, я создал класс, наследующий от tf.keras.Model). Поэтому вместо этого я попытался создать экземпляр новой модели, а затем просто загрузить веса, используя

model = myModel(someArgs)
model.load_weights(dirpath/variables)

Однако я получаю следующее сообщение об ошибке:

OSError: Невозможно открыть файл (невозможно открыть файл: name = 'dirpath / variables', errno = 13, сообщение об ошибке = 'Permission denied', flags = 0, o_flags = 0)

Итак, как я могу загрузить веса на модель? Файлы есть, я просто не знаю, как вернуть их в мою модель.

Answer 1

Разобрался, я ошибся путем. Мне нужно сделать model.load_weights(dirpath/variables/variables). Там есть два файла с именем variables с разными расширениями (.data-00000-of-00001 и .index), и это имя, которое вы хотите назвать.