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

Internal change

PiperOrigin-RevId: 433881862
parent 162c43c2
...@@ -16,7 +16,7 @@ ...@@ -16,7 +16,7 @@
from absl import logging from absl import logging
from official.nlp.xlnet import data_utils from official.legacy.xlnet import data_utils
SEG_ID_A = 0 SEG_ID_A = 0
SEG_ID_B = 1 SEG_ID_B = 1
......
...@@ -26,8 +26,8 @@ import numpy as np ...@@ -26,8 +26,8 @@ import numpy as np
import tensorflow as tf import tensorflow as tf
import sentencepiece as spm import sentencepiece as spm
from official.nlp.xlnet import classifier_utils from official.legacy.xlnet import classifier_utils
from official.nlp.xlnet import preprocess_utils from official.legacy.xlnet import preprocess_utils
flags.DEFINE_bool( flags.DEFINE_bool(
......
...@@ -28,7 +28,7 @@ import numpy as np ...@@ -28,7 +28,7 @@ import numpy as np
import tensorflow.compat.v1 as tf import tensorflow.compat.v1 as tf
import sentencepiece as spm import sentencepiece as spm
from official.nlp.xlnet import preprocess_utils from official.legacy.xlnet import preprocess_utils
FLAGS = flags.FLAGS FLAGS = flags.FLAGS
......
...@@ -25,7 +25,7 @@ from absl import logging ...@@ -25,7 +25,7 @@ from absl import logging
import tensorflow as tf import tensorflow as tf
import sentencepiece as spm import sentencepiece as spm
from official.nlp.xlnet import squad_utils from official.legacy.xlnet import squad_utils
flags.DEFINE_integer( flags.DEFINE_integer(
"num_proc", default=1, help="Number of preprocessing processes.") "num_proc", default=1, help="Number of preprocessing processes.")
......
...@@ -24,12 +24,12 @@ import numpy as np ...@@ -24,12 +24,12 @@ import numpy as np
import tensorflow as tf import tensorflow as tf
# pylint: disable=unused-import # pylint: disable=unused-import
from official.common import distribute_utils from official.common import distribute_utils
from official.nlp.xlnet import common_flags from official.legacy.xlnet import common_flags
from official.nlp.xlnet import data_utils from official.legacy.xlnet import data_utils
from official.nlp.xlnet import optimization from official.legacy.xlnet import optimization
from official.nlp.xlnet import training_utils from official.legacy.xlnet import training_utils
from official.nlp.xlnet import xlnet_config from official.legacy.xlnet import xlnet_config
from official.nlp.xlnet import xlnet_modeling as modeling from official.legacy.xlnet import xlnet_modeling as modeling
flags.DEFINE_integer("n_class", default=2, help="Number of classes.") flags.DEFINE_integer("n_class", default=2, help="Number of classes.")
flags.DEFINE_string( flags.DEFINE_string(
......
...@@ -24,12 +24,12 @@ from absl import logging ...@@ -24,12 +24,12 @@ from absl import logging
import tensorflow as tf import tensorflow as tf
# pylint: disable=unused-import # pylint: disable=unused-import
from official.common import distribute_utils from official.common import distribute_utils
from official.nlp.xlnet import common_flags from official.legacy.xlnet import common_flags
from official.nlp.xlnet import data_utils from official.legacy.xlnet import data_utils
from official.nlp.xlnet import optimization from official.legacy.xlnet import optimization
from official.nlp.xlnet import training_utils from official.legacy.xlnet import training_utils
from official.nlp.xlnet import xlnet_config from official.legacy.xlnet import xlnet_config
from official.nlp.xlnet import xlnet_modeling as modeling from official.legacy.xlnet import xlnet_modeling as modeling
flags.DEFINE_integer( flags.DEFINE_integer(
"num_predict", "num_predict",
......
...@@ -28,13 +28,13 @@ import tensorflow as tf ...@@ -28,13 +28,13 @@ import tensorflow as tf
# pylint: disable=unused-import # pylint: disable=unused-import
import sentencepiece as spm import sentencepiece as spm
from official.common import distribute_utils from official.common import distribute_utils
from official.nlp.xlnet import common_flags from official.legacy.xlnet import common_flags
from official.nlp.xlnet import data_utils from official.legacy.xlnet import data_utils
from official.nlp.xlnet import optimization from official.legacy.xlnet import optimization
from official.nlp.xlnet import squad_utils from official.legacy.xlnet import squad_utils
from official.nlp.xlnet import training_utils from official.legacy.xlnet import training_utils
from official.nlp.xlnet import xlnet_config from official.legacy.xlnet import xlnet_config
from official.nlp.xlnet import xlnet_modeling as modeling from official.legacy.xlnet import xlnet_modeling as modeling
flags.DEFINE_string( flags.DEFINE_string(
"test_feature_path", default=None, help="Path to feature of test set.") "test_feature_path", default=None, help="Path to feature of test set.")
......
...@@ -32,8 +32,8 @@ import numpy as np ...@@ -32,8 +32,8 @@ import numpy as np
import six import six
import tensorflow as tf import tensorflow as tf
from official.nlp.xlnet import data_utils from official.legacy.xlnet import data_utils
from official.nlp.xlnet import preprocess_utils from official.legacy.xlnet import preprocess_utils
SPIECE_UNDERLINE = u"▁" SPIECE_UNDERLINE = u"▁"
......
...@@ -22,7 +22,7 @@ from absl import logging ...@@ -22,7 +22,7 @@ from absl import logging
import tensorflow as tf import tensorflow as tf
from official.legacy.bert import model_training_utils from official.legacy.bert import model_training_utils
from official.nlp.xlnet import data_utils from official.legacy.xlnet import data_utils
# pytype: disable=attribute-error # pytype: disable=attribute-error
# pylint: disable=g-bare-generic,unused-import # pylint: disable=g-bare-generic,unused-import
......
...@@ -18,9 +18,8 @@ import copy ...@@ -18,9 +18,8 @@ import copy
import warnings import warnings
import tensorflow as tf import tensorflow as tf
from official.legacy.xlnet import data_utils
from official.nlp.modeling import networks from official.nlp.modeling import networks
from official.nlp.xlnet import data_utils
def gelu(x): def gelu(x):
......
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