module ‘tensorflow.python.training.experimental.mixed_precision‘ has no attribute ‘_register_wrapper

报错：

File "train.py", line 13, in <module>
    import itertools,tqdm,cv2,keras
  File "/home/cylab/Public/tangy/env3/lib/python3.6/site-packages/keras/__init__.py", line 25, in <module>
    from keras import models
  File "/home/cylab/Public/tangy/env3/lib/python3.6/site-packages/keras/models.py", line 20, in <module>
    from keras import metrics as metrics_module
  File "/home/cylab/Public/tangy/env3/lib/python3.6/site-packages/keras/metrics.py", line 27, in <module>
    from keras import activations
  File "/home/cylab/Public/tangy/env3/lib/python3.6/site-packages/keras/activations.py", line 20, in <module>
    from keras.layers import advanced_activations
  File "/home/cylab/Public/tangy/env3/lib/python3.6/site-packages/keras/layers/__init__.py", line 24, in <module>
    from keras.engine.input_layer import Input
  File "/home/cylab/Public/tangy/env3/lib/python3.6/site-packages/keras/engine/input_layer.py", line 21, in <module>
    from keras.engine import base_layer
  File "/home/cylab/Public/tangy/env3/lib/python3.6/site-packages/keras/engine/base_layer.py", line 41, in <module>
    from keras.mixed_precision import loss_scale_optimizer
  File "/home/cylab/Public/tangy/env3/lib/python3.6/site-packages/keras/mixed_precision/loss_scale_optimizer.py", line 1180, in <module>
    mixed_precision._register_wrapper_optimizer_cls(optimizer_v2.OptimizerV2,
AttributeError: module 'tensorflow.python.training.experimental.mixed_precision' has no attribute '_register_wrapper_optimizer_cls'

找到两个源文件：
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中的改为大写即可