首页 > 解决方案 > 从检查点加载 keras 权重

问题描述

这个问题有点傻!我正在训练一个模型,当我保存检查点时,我刚刚将 checkpoint_path 设置为: os.path.join(train_output_dir, "checkpoint/") 结果是:

.index
checkpoint
.data-00000-of-00001

我怎样才能恢复权重!我尝试 renmae.data到,weights.data但它不起作用! load_weights("checkpoint/")似乎没有加载正确的重量!

请注意,这.data是一个 1.8 G 文件,因此它没有损坏

标签: tensorflowkeras

解决方案


推荐阅读