AttributeError: module ‘tensorflow...mixed_precision‘ has no attribute ‘_register_wrapper

解决问题：

解决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