Commit aedb9802 authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

Move XLNet config and modeling code to xlnet folder.

PiperOrigin-RevId: 280720540
parent 98f10421
...@@ -27,12 +27,12 @@ from absl import logging ...@@ -27,12 +27,12 @@ from absl import logging
import numpy as np import numpy as np
import tensorflow as tf import tensorflow as tf
# pylint: disable=unused-import # pylint: disable=unused-import
from official.nlp import xlnet_config
from official.nlp import xlnet_modeling as modeling
from official.nlp.xlnet import common_flags from official.nlp.xlnet import common_flags
from official.nlp.xlnet import data_utils from official.nlp.xlnet import data_utils
from official.nlp.xlnet import optimization from official.nlp.xlnet import optimization
from official.nlp.xlnet import training_utils from official.nlp.xlnet import training_utils
from official.nlp.xlnet import xlnet_config
from official.nlp.xlnet import xlnet_modeling as modeling
from official.utils.misc import tpu_lib from official.utils.misc import tpu_lib
flags.DEFINE_integer("n_class", default=2, help="Number of classes.") flags.DEFINE_integer("n_class", default=2, help="Number of classes.")
......
...@@ -27,12 +27,12 @@ from absl import flags ...@@ -27,12 +27,12 @@ from absl import flags
from absl import logging from absl import logging
import tensorflow as tf import tensorflow as tf
# pylint: disable=unused-import # pylint: disable=unused-import
from official.nlp import xlnet_config
from official.nlp import xlnet_modeling as modeling
from official.nlp.xlnet import common_flags from official.nlp.xlnet import common_flags
from official.nlp.xlnet import data_utils from official.nlp.xlnet import data_utils
from official.nlp.xlnet import optimization from official.nlp.xlnet import optimization
from official.nlp.xlnet import training_utils from official.nlp.xlnet import training_utils
from official.nlp.xlnet import xlnet_config
from official.nlp.xlnet import xlnet_modeling as modeling
from official.utils.misc import tpu_lib from official.utils.misc import tpu_lib
flags.DEFINE_integer( flags.DEFINE_integer(
......
...@@ -31,13 +31,13 @@ from absl import logging ...@@ -31,13 +31,13 @@ from absl import logging
import tensorflow as tf import tensorflow as tf
# pylint: disable=unused-import # pylint: disable=unused-import
import sentencepiece as spm import sentencepiece as spm
from official.nlp import xlnet_config
from official.nlp import xlnet_modeling as modeling
from official.nlp.xlnet import common_flags from official.nlp.xlnet import common_flags
from official.nlp.xlnet import data_utils from official.nlp.xlnet import data_utils
from official.nlp.xlnet import optimization from official.nlp.xlnet import optimization
from official.nlp.xlnet import squad_utils from official.nlp.xlnet import squad_utils
from official.nlp.xlnet import training_utils from official.nlp.xlnet import training_utils
from official.nlp.xlnet import xlnet_config
from official.nlp.xlnet import xlnet_modeling as modeling
from official.utils.misc import tpu_lib from official.utils.misc import tpu_lib
flags.DEFINE_string( flags.DEFINE_string(
......
...@@ -13,7 +13,6 @@ ...@@ -13,7 +13,6 @@
# limitations under the License. # limitations under the License.
# ============================================================================== # ==============================================================================
"""XLNet training utils.""" """XLNet training utils."""
from __future__ import absolute_import from __future__ import absolute_import
from __future__ import division from __future__ import division
# from __future__ import google_type_annotations # from __future__ import google_type_annotations
...@@ -27,10 +26,11 @@ from absl import logging ...@@ -27,10 +26,11 @@ from absl import logging
# pytype: disable=attribute-error # pytype: disable=attribute-error
# pylint: disable=g-bare-generic,unused-import # pylint: disable=g-bare-generic,unused-import
import tensorflow as tf import tensorflow as tf
from typing import Any, Callable, Dict, Text, Optional
from official.modeling import model_training_utils from official.modeling import model_training_utils
from official.nlp.xlnet import data_utils from official.nlp.xlnet import data_utils
from official.nlp import xlnet_modeling as modeling from official.nlp.xlnet import xlnet_modeling as modeling
from typing import Any, Callable, Dict, Text, Optional
_MIN_SUMMARY_STEPS = 10 _MIN_SUMMARY_STEPS = 10
......
...@@ -20,7 +20,7 @@ from absl import logging ...@@ -20,7 +20,7 @@ from absl import logging
import numpy as np import numpy as np
import tensorflow as tf import tensorflow as tf
from official.nlp import xlnet_modeling from official.nlp.xlnet import xlnet_modeling
class PositionalEmbeddingLayerTest(tf.test.TestCase): class PositionalEmbeddingLayerTest(tf.test.TestCase):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment