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
import numpy as np
import tensorflow as tf
# 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 data_utils
from official.nlp.xlnet import optimization
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
flags.DEFINE_integer("n_class", default=2, help="Number of classes.")
......
......@@ -27,12 +27,12 @@ from absl import flags
from absl import logging
import tensorflow as tf
# 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 data_utils
from official.nlp.xlnet import optimization
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
flags.DEFINE_integer(
......
......@@ -31,13 +31,13 @@ from absl import logging
import tensorflow as tf
# pylint: disable=unused-import
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 data_utils
from official.nlp.xlnet import optimization
from official.nlp.xlnet import squad_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
flags.DEFINE_string(
......
......@@ -13,7 +13,6 @@
# limitations under the License.
# ==============================================================================
"""XLNet training utils."""
from __future__ import absolute_import
from __future__ import division
# from __future__ import google_type_annotations
......@@ -27,10 +26,11 @@ from absl import logging
# pytype: disable=attribute-error
# pylint: disable=g-bare-generic,unused-import
import tensorflow as tf
from typing import Any, Callable, Dict, Text, Optional
from official.modeling import model_training_utils
from official.nlp.xlnet import data_utils
from official.nlp import xlnet_modeling as modeling
from typing import Any, Callable, Dict, Text, Optional
from official.nlp.xlnet import xlnet_modeling as modeling
_MIN_SUMMARY_STEPS = 10
......
......@@ -20,7 +20,7 @@ from absl import logging
import numpy as np
import tensorflow as tf
from official.nlp import xlnet_modeling
from official.nlp.xlnet import xlnet_modeling
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