                                 self.config["learning_rate"],
                                 self.config.get("clipnorm", False),
                                 self._args).optimizer
          if self._settings.use_mixed_precision:
              optimizer = self._settings.LossScaleOptimizer(optimizer, loss_scale="dynamic")
          if get_backend() == "amd":
              self._rewrite_plaid_outputs()
          self._loss.configure(self._model)
          self._model.compile(optimizer=optimizer, loss=self._loss.functions)
          if not self._is_predict:
 
</s> remove             self._mixed_precision = tf.keras.mixed_precision.experimental
 </s> add             self._mixed_precision = tf.keras.mixed_precision </s> remove             from tensorflow.python.keras.mixed_precision.experimental import \
 </s> add             from tensorflow.python.keras.mixed_precision import \ </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)
            optimizer = self._settings.LossScaleOptimizer(optimizer, True)
                                 self.config["learning_rate"],
                                 self.config.get("clipnorm", False),
                                 self._args).optimizer
          if self._settings.use_mixed_precision:
            optimizer = self._settings.LossScaleOptimizer(optimizer, True)
          if get_backend() == "amd":
              self._rewrite_plaid_outputs()
          self._loss.configure(self._model)
          self._model.compile(optimizer=optimizer, loss=self._loss.functions)
          if not self._is_predict:
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