keras载入与保存模型
载入模型
import os
if(os.path.exists('my_model.h5')):
network = models.load_model('my_model.h5')
print('已载入网络')
else:
# 架构神经网络
保存模型
network.save('my_model.h5')
keras载入与保存模型
载入模型
import os
if(os.path.exists('my_model.h5')):
network = models.load_model('my_model.h5')
print('已载入网络')
else:
# 架构神经网络
保存模型
network.save('my_model.h5')