Commit c13243c1 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Internal change

PiperOrigin-RevId: 411638103
parent 05b8eaf2
...@@ -28,7 +28,7 @@ from official.nlp.configs import encoders ...@@ -28,7 +28,7 @@ from official.nlp.configs import encoders
from official.nlp.data import pretrain_dataloader from official.nlp.data import pretrain_dataloader
from official.nlp.modeling import layers from official.nlp.modeling import layers
from official.nlp.modeling import models from official.nlp.modeling import models
from official.nlp.projects.mobilebert import distillation from official.projects.mobilebert import distillation
class DistillationTest(tf.test.TestCase, parameterized.TestCase): class DistillationTest(tf.test.TestCase, parameterized.TestCase):
......
...@@ -18,7 +18,7 @@ from absl import flags ...@@ -18,7 +18,7 @@ from absl import flags
from absl import logging from absl import logging
import tensorflow as tf import tensorflow as tf
from official.nlp.projects.mobilebert import model_utils from official.projects.mobilebert import model_utils
FLAGS = flags.FLAGS FLAGS = flags.FLAGS
......
...@@ -31,7 +31,7 @@ from official.modeling import performance ...@@ -31,7 +31,7 @@ from official.modeling import performance
from official.modeling.fast_training.progressive import train_lib from official.modeling.fast_training.progressive import train_lib
from official.modeling.fast_training.progressive import trainer as prog_trainer_lib from official.modeling.fast_training.progressive import trainer as prog_trainer_lib
from official.nlp.data import pretrain_dataloader from official.nlp.data import pretrain_dataloader
from official.nlp.projects.mobilebert import distillation from official.projects.mobilebert import distillation
FLAGS = flags.FLAGS FLAGS = flags.FLAGS
......
...@@ -21,7 +21,7 @@ from absl import logging ...@@ -21,7 +21,7 @@ from absl import logging
import numpy as np import numpy as np
import tensorflow.compat.v1 as tf import tensorflow.compat.v1 as tf
from official.nlp.projects.mobilebert import model_utils from official.projects.mobilebert import model_utils
FLAGS = flags.FLAGS FLAGS = flags.FLAGS
......
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