              device_compatibility_check._logged_compatibility_check = True
              logger.debug("New value: %s", device_compatibility_check._logged_compatibility_check)
  
          policy = self._mixed_precision.Policy('mixed_float16')
          self._mixed_precision.set_policy(policy)
          logger.debug("Enabled mixed precision. (Compute dtype: %s, variable_dtype: %s)",
                       policy.compute_dtype, policy.variable_dtype)
          return True
  
      @classmethod
 
</s> remove             from tensorflow.python.keras.mixed_precision.experimental import \
 </s> add             from tensorflow.python.keras.mixed_precision import \ </s> remove             self._mixed_precision = tf.keras.mixed_precision.experimental
 </s> add             self._mixed_precision = tf.keras.mixed_precision </s> remove         max_ver = 2.2
 </s> add         max_ver = 2.4 </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
        self._mixed_precision.set_global_policy(policy)
              device_compatibility_check._logged_compatibility_check = True
              logger.debug("New value: %s", device_compatibility_check._logged_compatibility_check)
  
          policy = self._mixed_precision.Policy('mixed_float16')
        self._mixed_precision.set_global_policy(policy)
          logger.debug("Enabled mixed precision. (Compute dtype: %s, variable_dtype: %s)",
                       policy.compute_dtype, policy.variable_dtype)
          return True
  
      @classmethod
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