若是遇到如题错误,则将

keras.backend.tensorflow_backend.set_session(tf.compat.v1.Session(config=config))

改为

tensorflow
一个面向所有人的开源机器学习框架

tf.compat.v1.keras.backend.set_session(tf.compat.v1.Session(config=config))

即可

参考来源:https://www.tensorflow.org/api_docs/python/tf/compat/v1/keras/backend/set_session

GitHub 加速计划 / te / tensorflow
20
4
下载
一个面向所有人的开源机器学习框架
最近提交(Master分支:1 个月前 )
4f64a3d5 Instead, check for this case in `ResolveUsers` and `ResolveOperand`, by querying whether the `fused_expression_root` is part of the `HloFusionAdaptor`. This prevents us from stepping into nested fusions. PiperOrigin-RevId: 724311958 1 个月前
aa7e952e Fix a bug in handling negative strides, and add a test case that exposes it. We can have negative strides that are not just -1, e.g. with a combining reshape. PiperOrigin-RevId: 724293790 1 个月前
Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐