AttributeError: module '...mixed_precision' has no attribute '_register_wrapper_optimizer_cls'

Question

##解决tensorflow2.0安装后，出现如下报错方法

AttributeError: module 'tensorflow.python.training.experimental.mixed_precision' has no attribute '_register_wrapper_optimizer_cls'

错误如图：

我的解答思路和尝试过的方法

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