tensorflow2: как сохранить и загрузить вес модели в/из памяти?

Как сохранить и загрузить веса модели в/из памяти вместо диска в tensorflow2.0?


person Leon    schedule 10.02.2020    source источник


Ответы (1)


Я не думаю, что вы можете сохранить и загрузить модель в "память/ОЗУ". Хотя после того, как вы закончите обучение, экземпляр обученной модели все еще останется в памяти, пока программа не завершится, что вы можете использовать для прогнозирования и т. д. Вы можете попробовать Google Colab, где экземпляр можно использовать в течение более длительного времени (прибл. 12 часов). Поэтому, как только вы обучите модель, вы можете использовать ее много раз для прогнозирования и т. д.

Для сохранения и загрузки вы должны предоставить ему некоторое постоянное пространство памяти (ПЗУ).

person Rishabh Sahrawat    schedule 10.02.2020