Move modeling_training_utils.py to bert folder

PiperOrigin-RevId: 303050809

official/modeling/model_training_utils_test.py → official/nlp/bert/model_training_utils_test.py

@@ -28,7 +28,7 @@ import tensorflow as tf
 
 from tensorflow.python.distribute import combinations
 from tensorflow.python.distribute import strategy_combinations
-from official.modeling import model_training_utils
+from official.nlp.bert import model_training_utils
 
 
 def eager_strategy_combinations():

official/nlp/bert/run_classifier.py

@@ -25,8 +25,6 @@ from absl import app
 from absl import flags
 from absl import logging
 import tensorflow as tf
-
-from official.modeling import model_training_utils
 from official.modeling import performance
 from official.nlp import optimization
 from official.nlp.bert import bert_models
@@ -34,6 +32,7 @@ from official.nlp.bert import common_flags
 from official.nlp.bert import configs as bert_configs
 from official.nlp.bert import input_pipeline
 from official.nlp.bert import model_saving_utils
+from official.nlp.bert import model_training_utils
 from official.utils.misc import distribution_utils
 from official.utils.misc import keras_utils
 

official/nlp/bert/run_pretraining.py

@@ -22,14 +22,13 @@ from absl import flags
 from absl import logging
 import gin
 import tensorflow as tf
-
-from official.modeling import model_training_utils
 from official.modeling import performance
 from official.nlp import optimization
 from official.nlp.bert import bert_models
 from official.nlp.bert import common_flags
 from official.nlp.bert import configs
 from official.nlp.bert import input_pipeline
+from official.nlp.bert import model_training_utils
 from official.utils.misc import distribution_utils
 
 

official/nlp/bert/run_squad_helper.py

@@ -23,14 +23,13 @@ import os
 from absl import flags
 from absl import logging
 import tensorflow as tf
-
-from official.modeling import model_training_utils
 from official.modeling import performance
 from official.nlp import optimization
 from official.nlp.bert import bert_models
 from official.nlp.bert import common_flags
 from official.nlp.bert import input_pipeline
 from official.nlp.bert import model_saving_utils
+from official.nlp.bert import model_training_utils
 from official.nlp.bert import squad_evaluate_v1_1
 from official.nlp.data import squad_lib_sp
 from official.utils.misc import keras_utils

official/nlp/xlnet/training_utils.py

@@ -28,7 +28,7 @@ from absl import logging
 import tensorflow as tf
 from typing import Any, Callable, Dict, Text, Optional
 
-from official.modeling import model_training_utils
+from official.nlp.bert import model_training_utils
 from official.nlp.xlnet import data_utils
 from official.nlp.xlnet import xlnet_modeling as modeling