TensorFlow加载模型权重报错：未找到匹配文件求助

问题核心 #

报错显示找不到路径./model_save.\\encoder_weights，根源是代码中路径定义错误，加上可能的权重文件未正确生成。

解决方案 #

1. 修正路径中的多余符号
   代码里save_dir = './model_save.'末尾多了一个英文句号，导致生成的路径不符合预期。将其修改为：

   save_dir = './model_save'
   

2. 确认权重文件存在且路径匹配

   • 检查./model_save目录下是否存在以encoder_weights和decoder_weights为前缀的文件（TensorFlow保存权重会生成.index、.data-*等多个文件）。
   • 如果目录不存在，先创建目录再保存权重：

     import os
     save_dir = './model_save'
     if not os.path.exists(save_dir):
         os.makedirs(save_dir)
     # 执行保存操作（首次运行必须先保存才能加载）
     encoder.save_weights(os.path.join(save_dir, "encoder_weights"))
     decoder.save_weights(os.path.join(save_dir, "decoder_weights"))
     

3. 保证保存与加载路径一致
   保存权重时使用的路径要和加载时完全相同，避免相对路径因当前工作目录变化导致的问题。

修正后的完整代码 #

import os
save_dir = './model_save'
encoder_prefix = os.path.join(save_dir, "encoder_weights")
decoder_prefix = os.path.join(save_dir, "decoder_weights")

# 创建保存目录（如果不存在）
if not os.path.exists(save_dir):
    os.makedirs(save_dir)

# 先保存权重（首次运行需执行）
encoder.save_weights(encoder_prefix)
decoder.save_weights(decoder_prefix)

# 加载权重
encoder.load_weights(encoder_prefix)
decoder.load_weights(decoder_prefix)

内容的提问来源于stack exchange，提问作者fovad rasoli