python - Tensorflow 给出“ValueError：检查输入时出错”

Question

我正在尝试使用 Deep Q-Network 代理解决 OpenAI 健身房 Breakout-V0。
每次当我的代理到达以下地步时：

• replay_memory 已填满，可以开始训练
• 首次达到 copy_target_network 间隔
• target_network 第一次预测

Tensorflow 抛出以下错误：

Error when checking input: expected dense_3_input to have shape (33600,) but got array with shape (1,)

state当我在调用之前仅 1 行打印传入数组的形状时predict(state)，它确认的形状state是(33600,)
在显示此错误之前，模型能够predict_on_batch()在训练循环中使用完全相同的数据（但已批处理）

有谁知道如何解决这个问题？如果我遗漏任何信息，我很乐意提供更多详细信息和信息

版本：
Python 3.8.7
TensorFlow 2.4.1
Gym 0.18.0

标签： pythontensorflowdeep-learningreinforcement-learning

Answer 1

正如史努比博士所说，这是一个简单的解决方案

只好做np.reshape(state, (1, 33600))