Tensorflow保存权重InvalidArgumentError

Question

使用以下最少的代码

import tensorflow as tf

model = tf.keras.models.Sequential([
  tf.keras.layers.Flatten(input_shape=(28, 28)),
  tf.keras.layers.Dense(128, activation='relu'),
  tf.keras.layers.Dropout(0.2),
  tf.keras.layers.Dense(10)
])

model.save_weights('my_checkpoint')

我收到以下错误（我翻译的最后一部分是用我的母语打印的）

InvalidArgumentError: {{function_node __wrapped__SaveV2_dtypes_5_device_/job:localhost/replica:0/task:0/device:CPU:0}} Failed to
create a NewWriteableFile: my_checkpoint_temp/part-00000-of-00001.data-00000-of-00001.tempstate15549814337234001039 : The syntax 
of the file name, folder name or volume name is incorrect.
; no protocol option [Op:SaveV2]

这怎么可能？我正在使用张量流2.14.0

Answer 1

InvalidArgumentError：{{function_node _wrapped__SaveV2_dtypes_5_device/job:localhost/replica:0/task:0/device:CPU:0}} 无法创建 NewWriteableFile: my_checkpoint_temp/part-00000-of-00001.data-00000-of-00001.tempstate15549814337234001039 ：文件名、文件夹名或卷名的语法为 不正确。 ;无协议选项 [Op:SaveV2]

问题似乎与尝试存储模型权重时的文件路径或名称有关。该错误消息特别强调了文件名“my_checkpoint_temp/part-00000-of-00001.data-00000-of-00001.tempstate15549814337234001039.

”的问题。

确保提供有效的文件路径来保存权重

示例：-

model.save_weights('/path/to/your/directory/my_checkpoint')