tensors = pywrap_tfe.TFE_Py_Execute(ctx._handle, device_name, op_name, tensorflow.python.framework.errors_impl.InvalidArgumentError: lr is not a scalar : [3] [[node Adam/Adam/update_7/ResourceApplyAdam (defined at F:\KXZB\PSO_RNN.py:157) ]] [Op:__inference_train_function_2112]
时间: 2024-03-30 16:33:29 浏览: 373
TensorFlow.js、迁移学习与AI产品创新之道
根据错误提示,似乎是在执行 TensorFlow 的 Adam 优化器时出现了问题,原因是 lr(学习率)不是一个标量。请检查你的代码中是否有类似于 lr = [3] 这样的问题,确保学习率是一个标量值。另外,也可以检查一下你使用的优化器的参数是否正确设置。如果问题仍然存在,请提供更多的代码细节以便排查问题。
阅读全文