          self._set_tf_settings(allow_growth, arguments.exclude_gpus)
  
          use_mixed_precision = not is_predict and mixed_precision and get_backend() == "nvidia"
          if use_mixed_precision:
              self._mixed_precision = tf.keras.mixed_precision.experimental
          else:
              self._mixed_precision = None
  
          self._use_mixed_precision = self._set_keras_mixed_precision(use_mixed_precision,
                                                                      bool(arguments.exclude_gpus))
 
</s> remove             optimizer = self._settings.LossScaleOptimizer(optimizer, loss_scale="dynamic")
 </s> add             optimizer = self._settings.LossScaleOptimizer(optimizer, True) </s> remove             If Tensorflow is not found, or is not between versions 2.2 and 2.2
 </s> add             If Tensorflow is not found, or is not between versions 2.2 and 2.4 </s> remove         max_ver = 2.2
 </s> add         max_ver = 2.4 </s> remove         self._mixed_precision.set_policy(policy)
 </s> add         self._mixed_precision.set_global_policy(policy) </s> remove             from tensorflow.python.keras.mixed_precision.experimental import \
 </s> add             from tensorflow.python.keras.mixed_precision import \
            self._mixed_precision = tf.keras.mixed_precision
          self._set_tf_settings(allow_growth, arguments.exclude_gpus)
  
          use_mixed_precision = not is_predict and mixed_precision and get_backend() == "nvidia"
          if use_mixed_precision:
            self._mixed_precision = tf.keras.mixed_precision
          else:
              self._mixed_precision = None
  
          self._use_mixed_precision = self._set_keras_mixed_precision(use_mixed_precision,
                                                                      bool(arguments.exclude_gpus))
keep keep keep keep replace keep keep keep keep keep
 Update tensorflow to 2.4.0rc4 (#1086)

* Update tensorflow to 2.4.0rc1

* Update requirements_nvidia.txt

* Update requirements_nvidia.txt 
plugins/train/model/_base.py