Tensorflow保存模型，加载模型

最新推荐文章于 2024-02-25 10:04:18 发布

神经学徒小周

最新推荐文章于 2024-02-25 10:04:18 发布

阅读量202

点赞数 1

文章标签： tensorflow 深度学习

本文链接：https://blog.csdn.net/z1445056258/article/details/111144238

版权

Tensorflow保存模型，加载模型

#保存模型并删除：
model.save_weights('/content/data/cifar10_weights.ckpt')
del model

#加载模型并评估
model = My_Net() #实例化与之前的模型一样，不然参数不匹配
model.compile(optimizer=tf.optimizers.Adam(lr=1e-3),
              loss=tf.losses.CategoricalCrossentropy(from_logits=True),
              metrics=['accuracy'])
model.load_weights('/content/data/cifar10_weights.ckpt')
model.evaluate(test_db)