Deep Learning/tensorflow

[tensorflow] model save, load

heedy 2022. 11. 29. 17:00
728x90

- model save

import tensorflow as tf

model.save('model.h5')

 

- model load

import tensorflow as tf

model = tf.keras.models.load_model('model.h5')
728x90