重新训练后加载Keras模型时出现AttributeError：'Ones'对象没有'_set_dtype_policy'属性

我遇到了一个TensorFlow的诡异行为问题，实在需要大家帮忙分析。具体的场景流程是这样的：

• 用Gradient Tape训练模型
• 执行model.save(my_dir)保存模型
• 执行tf.keras.models.load_model(my_dir, compile=False)加载模型
• 对加载后的模型进行重新训练
• 执行model.save(my_dir1)保存重新训练后的模型
• 执行tf.keras.models.load_model(my_dir1, compile=False)时触发AttributeError

错误堆栈信息： #

self.model = tf.keras.models.load_model(my_dir1, compile=False)
  File "/usr/local/lib/python3.8/dist-packages/keras/utils/traceback_utils.py", line 70, in error_handler
    raise e.with_traceback(filtered_tb) from None
  File "/usr/local/lib/python3.8/dist-packages/keras/saving/legacy/saved_model/load.py", line 625, in _revive_layer_or_model_from_config
    obj._set_dtype_policy(metadata["dtype"])
AttributeError: 'Ones' object has no attribute '_set_dtype_policy'

目前我只在某个技术社区讨论里看到过类似问题的提及，但还没找到有效的解决办法。

备注：内容来源于stack exchange，提问作者user25345530