Как перенести обучение с tensorflow 1.14 на tf 2?

У меня есть собственная обученная модель (обученная в tf 1.14), которую я использую для предварительной подготовки другой модели. Таким образом, это не одна из известных моделей, таких как ResNet/и т. д., которые вы можете найти в tf.keras.applications.

Я хочу переключить свою работу на tf 2.0, но не могу найти ссылку, чтобы предоставить мне подробную информацию о том, как загрузить контрольные точки tf 1.14 и график в tf2.

Возможно ли вообще передавать-учиться в tensorflow 2, используя информацию о модели из tf 1?


migration tensorflow graph transfer-learning checkpoint
person n_esk    schedule 20.11.2019    source источник

Ответы (1)


arrow_upward
0
arrow_downward

Это должно быть возможно, если вы:

  • сохранить веса из модели tf 1.14 (old_model.save_weights(путь))
  • построить ту же модель в 2.0 (только макет слоя) (new_model = build_and_compile_model())
  • загрузить веса 1.14 в модель 2.0 (new_model.load_weights(path/to/old_model_weights))

надеюсь, это поможет

person Flo    schedule 25.02.2020