解决问题:
解决tensorflow2.0安装后,出现如下报错现象
已尝试过的解决方案
1、找到两个源文件:
keras里的
# pylint: disable=protected-access
mixed_precision._register_wrapper_optimizer_cls(optimizer_v2.OptimizerV2,
LossScaleOptimizerV1)
tensorflow里的:
_REGISTERED_WRAPPER_OPTIMIZER_CLS = {
optimizer.Optimizer:
loss_scale_optimizer_v1.MixedPrecisionLossScaleOptimizer,
}
解决方法:对比发现是因为大小写导致的,把keras中的改为大写即可
2、版本匹配问题,使用如下指令更新keras库
pip3 uninstall keras
pip3 install --upgrade keras
自己的解决方案:
根据错误提示,找到mixed_precision.py程序并打开,在程序末加入以下语句,问题解决。
def _register_wrapper_optimizer_cls(optimizer_cls, wrapper_optimizer_cls):
_REGISTERED_WRAPPER_OPTIMIZER_CLS[optimizer_cls] = wrapper_optimizer_cls