高效码农

RNN tensorflow 集锦问题汇总

一、建Embeddin层优化

1、报错信息:Unrecognized keyword arguments passed to Embedding:{'batch_input_shape': [64, None]}

2、模型配置可优化

在创建Embedding层时,使用了batch_size作为参数。这在某些情况下可以加速初始化过程,但不是必须的,且可能会导致在不同batch大小上重复使用模型时出现问题。通常,我们建议不指定batch_size,让Keras在运行时根据实际情况决定。

3、解决方案:移除batch_size参数

 # 创建模型
    model = tf.keras.Sequential([
        tf.keras.layers.Embedding(vocab_size, embedding_dim),
        tf.keras.layers.GRU(rnn_units, return_sequences=True, stateful=False, recurrent_initializer='glorot_uniform'),
        tf.keras.layers.Dense(vocab_size, activation=None)  # 可根据任务调整激活函数,默认为None
    ])

二、错误信息

When using save_weights_only=True in ModelCheckpoint, the filepath provided must end in .weights.h5 (Keras weights format). Received:

解决方案:

根据错误提示,当在 tf.keras.callbacks.ModelCheckpoint 中设置 save_weights_only=True 时,保存路径(filepath)应以 .weights.h5 结尾。为了修正这个问题,请修改检查点文件名格式。以下是修复后的代码:

# %% 
# 进行训练
# 检查点保存至的目录
checkpoint_dir = '/training_checkpoints'
# 检查点的文件名格式,确保以 .weights.h5 结尾
checkpoint_prefix = os.path.join(checkpoint_dir, "ckpt_{epoch:02d}.weights.h5")
# 训练的回调
checkpoint_callback = tf.keras.callbacks.ModelCheckpoint(filepath=checkpoint_prefix, save_weights_only=True)
# 进行训练
history = model.fit(dataset, epochs=20, callbacks=[checkpoint_callback])

当前页面是本站的「Google AMP」版。查看和发表评论请点击:完整版 »