fix determinism

Author: fsx950223

efficientdet/requirements.txt

@@ -5,8 +5,8 @@ numpy>=1.19.4
 Pillow>=6.0.0
 PyYAML>=5.1
 six>=1.15.0
-tensorflow>=2.7.0
-tensorflow-addons>=0.15
+tensorflow>=2.8.0
+tensorflow-addons>=0.16.1
 tensorflow-hub>=0.11
 neural-structured-learning>=1.3.1
 Cython>=0.29.13

efficientdet/tf2/train.py

@@ -164,11 +164,11 @@ def main(_):
       tf.config.experimental.set_memory_growth(gpu, True)
 
   if FLAGS.tf_random_seed:
-    tf.random.set_seed(FLAGS.tf_random_seed)
+    tf.keras.utils.set_random_seed(FLAGS.tf_random_seed)
+    tf.config.experimental.enable_op_determinism()
 
   if FLAGS.debug:
     tf.debugging.set_log_device_placement(True)
-    os.environ['TF_DETERMINISTIC_OPS'] = '1'
     logging.set_verbosity(logging.DEBUG)
 
   if FLAGS.strategy == 'tpu':
@@ -241,9 +241,11 @@ def get_dataset(is_training, config):
     else:
       model = train_lib.EfficientDetNetTrain(config=config)
     model = setup_model(model, config)
+
     if FLAGS.debug:
       tf.data.experimental.enable_debug_mode()
       tf.config.run_functions_eagerly(True)
+
     if tf.train.latest_checkpoint(FLAGS.model_dir):
       ckpt_path = tf.train.latest_checkpoint(FLAGS.model_dir)
       util_keras.restore_ckpt(